{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "\n",
    "os.environ['CUDA_VISIBLE_DEVICES'] = ''"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "from utils.copied_utils import (\n",
    "    compute_input_and_target_lengths,\n",
    "    DataCollatorForT5MLM,\n",
    "    tokenize_function,\n",
    "    DataCollatorForNI,\n",
    ")\n",
    "from streaming.base.format.mds.encodings import Encoding, _encodings\n",
    "from streaming import LocalDataset\n",
    "import torch\n",
    "import numpy as np\n",
    "\n",
    "\n",
    "class UInt16(Encoding):\n",
    "    def encode(self, obj) -> bytes:\n",
    "        return obj.tobytes()\n",
    "\n",
    "    def decode(self, data: bytes):\n",
    "        return np.frombuffer(data, np.uint16)\n",
    "\n",
    "\n",
    "_encodings['uint16'] = UInt16\n",
    "\n",
    "\n",
    "class DatasetFixed(torch.utils.data.Dataset):\n",
    "    def __init__(self, local):\n",
    "        self.dataset = LocalDataset(local=local)\n",
    "\n",
    "    def __getitem__(self, idx):\n",
    "        data = self.dataset[idx]\n",
    "        data.pop('token_type_ids', None)\n",
    "        for k in data.keys():\n",
    "            data[k] = data[k].astype(np.int64)\n",
    "        return data\n",
    "\n",
    "    def __len__(self):\n",
    "        return len(self.dataset)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "dataset = DatasetFixed(local='/home/ubuntu/mosaic-nanot5-512')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "(568, 114)"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "before_mask_input_length, target_length = compute_input_and_target_lengths(\n",
    "    inputs_length=512,\n",
    "    noise_density=0.15,\n",
    "    mean_noise_span_length=3.0,\n",
    ")\n",
    "before_mask_input_length, target_length"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.\n"
     ]
    }
   ],
   "source": [
    "from transformers import AutoTokenizer, AutoConfig\n",
    "\n",
    "tokenizer = AutoTokenizer.from_pretrained('out-base-1.1')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "config = AutoConfig.from_pretrained('out-base-1.1')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "data_collator = DataCollatorForT5MLM(\n",
    "    tokenizer=tokenizer,\n",
    "    noise_density=0.15,\n",
    "    mean_noise_span_length=3.0,\n",
    "    input_length=512,\n",
    "    target_length=114,\n",
    "    pad_token_id=config.pad_token_id,\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'input_ids': tensor([[    1,  3536,  3146,  ...,  2309, 32072,     2],\n",
       "         [ 9584,  3146, 18376,  ..., 10035, 32072,     2],\n",
       "         [  743,  2886,   790,  ...,   530, 32072,     2]]),\n",
       " 'labels': tensor([[32099,   227,   170,   101,   124, 18963,   101,   120, 32098,  1676,\n",
       "           2802,  2309, 32097,  2309,  7159,   295, 21329,    15, 32096, 29589,\n",
       "            352,  5090,  1810, 25920, 32095,    12, 32094,     5,   449,  6342,\n",
       "          32093,     5,   647, 32092,  2244, 11414, 32091,  2602,   313,  1305,\n",
       "           1427,  2200,   939, 32090,   352, 32089,   901,  2150,  2309, 11130,\n",
       "             15,  2309,  9404, 32088,  7159, 32087,  2309,  3146, 32086,  2309,\n",
       "           1627, 32085,  1206,  9584, 32084, 10404,   586,  3065,   275, 32083,\n",
       "           3146,  5777,    12,   313,   847,   327, 32082, 12849, 32081, 21847,\n",
       "            383, 32080,   287, 32079,    17,  1733, 20348,  8466, 32078,   758,\n",
       "            492, 32077,    17,  3536,  3146, 32076,  5378, 32075,   449,   526,\n",
       "          26147,   403,   918,  2309,    16,  9584, 32074,    81,  1943, 32073,\n",
       "          25637, 32072,    16,     2],\n",
       "         [32099, 25928,     5, 32098,   344,  2309, 32097,   282, 18009,  2309,\n",
       "           3146,  1083, 25666, 32096,  1512, 32095,   779,    17,  5495, 32094,\n",
       "           3146,  9982,   436, 32093,  1553, 32092, 18009,  2309, 32091,  2021,\n",
       "          32090,   492, 32089,  8716,   836,  1607, 14951, 32088,    15,   313,\n",
       "           2239,  2370,   918,  3462,   484,   446, 32087,  3146,  1350, 32086,\n",
       "          12445,  2741, 32085,  3146,   332, 32084,  2309,  3146,   332, 18902,\n",
       "            323,  2309, 32083,  1497, 12445, 32082,  5323, 32081,   323, 26625,\n",
       "          32080,   295,  6996,    17, 32079,  3694,   450, 32078, 10035, 32077,\n",
       "            323, 32076,   812,  8484,   881,  7203,  4583, 32075, 11015,   313,\n",
       "           9935,   304,    17,   330, 32074,  1049, 12041,    15,   313, 32073,\n",
       "          20207, 16692,   313, 11844,  3486, 12632,   403, 10667,   292,    85,\n",
       "            523, 32072,   449,     2],\n",
       "         [32099,    15,  1627,    17,  7556,   881,  7203,  6538,   352,    45,\n",
       "          32098,  1122, 32097,   273, 18462,  8484,  2699, 32096,  2732,   616,\n",
       "          32095,   313, 22761,   836, 32094,   338, 11039,  1655,    76, 32093,\n",
       "           3146, 32092,  2200,   939,   918, 32091,   500,  3763,    15,   313,\n",
       "          23567, 15650,   403,   449, 32090,  1313,  2462,   911, 32089,  3337,\n",
       "          17098,   787, 30217, 15650,   403,    17,  3536, 32088,   313, 19073,\n",
       "             17,  3536,  3146, 32087,  2103, 32086,  2512,   313,   807,  9306,\n",
       "          32085,   532, 32084,  2077,  4059,   295, 32083,  1490, 32082,  3065,\n",
       "            275,  2201, 32081,   305, 32080, 21847, 13081, 32079,   680, 32078,\n",
       "           1377,  4980, 32077,   447, 32076,    81, 32075,   766,   305,  4203,\n",
       "            352, 32074,   939,   918,  2309,    16,  9584, 32073,   309, 32072,\n",
       "           4225,    17,  3536,     2]])}"
      ]
     },
     "execution_count": 30,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "b = [dataset[i] for i in range(3)]\n",
    "b = data_collator(b)\n",
    "b"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [],
   "source": [
    "from transformers import T5ForConditionalGeneration\n",
    "from pytorch_lightning import LightningModule"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "metadata": {},
   "outputs": [],
   "source": [
    "class Module(LightningModule):\n",
    "    def __init__(self):\n",
    "        super().__init__()\n",
    "        config = AutoConfig.from_pretrained(\n",
    "            './out-base-1.1'\n",
    "        )\n",
    "        self.model = T5ForConditionalGeneration.from_pretrained(\n",
    "            './out-base-1.1',\n",
    "            config=config,\n",
    "        )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "total 8.4G\r\n",
      "-rw-r--r-- 1 ubuntu ubuntu 2.8G Apr 14 03:39 'model-epoch=00-step=1000.ckpt'\r\n",
      "-rw-r--r-- 1 ubuntu ubuntu 2.8G Apr 14 04:19 'model-epoch=00-step=2000.ckpt'\r\n",
      "-rw-r--r-- 1 ubuntu ubuntu 2.8G Apr 14 04:50 'model-epoch=00-step=3000.ckpt'\r\n"
     ]
    }
   ],
   "source": [
    "!ls -lh logs/base"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "metadata": {},
   "outputs": [],
   "source": [
    "model = Module()\n",
    "weights = model.state_dict()\n",
    "old_weights = torch.load('logs/base/model-epoch=00-step=3000.ckpt',\n",
    "                             map_location=torch.device('cpu'))['state_dict'].items()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "model.shared.weight model.shared.weight\n",
      "model.encoder.embed_tokens.weight model.encoder.embed_tokens.weight\n",
      "model.encoder.block.0.layer.0.SelfAttention.q.weight model.encoder.block.0.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.0.layer.0.SelfAttention.k.weight model.encoder.block.0.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.0.layer.0.SelfAttention.v.weight model.encoder.block.0.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.0.layer.0.SelfAttention.o.weight model.encoder.block.0.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.0.layer.0.SelfAttention.relative_attention_bias.weight model.encoder.block.0.layer.0.SelfAttention.relative_attention_bias.weight\n",
      "model.encoder.block.0.layer.0.layer_norm.weight model.encoder.block.0.layer.0.layer_norm.weight\n",
      "model.encoder.block.0.layer.1.DenseReluDense.wi_0.weight model.encoder.block.0.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.0.layer.1.DenseReluDense.wi_1.weight model.encoder.block.0.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.0.layer.1.DenseReluDense.wo.weight model.encoder.block.0.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.0.layer.1.layer_norm.weight model.encoder.block.0.layer.1.layer_norm.weight\n",
      "model.encoder.block.1.layer.0.SelfAttention.q.weight model.encoder.block.1.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.1.layer.0.SelfAttention.k.weight model.encoder.block.1.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.1.layer.0.SelfAttention.v.weight model.encoder.block.1.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.1.layer.0.SelfAttention.o.weight model.encoder.block.1.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.1.layer.0.layer_norm.weight model.encoder.block.1.layer.0.layer_norm.weight\n",
      "model.encoder.block.1.layer.1.DenseReluDense.wi_0.weight model.encoder.block.1.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.1.layer.1.DenseReluDense.wi_1.weight model.encoder.block.1.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.1.layer.1.DenseReluDense.wo.weight model.encoder.block.1.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.1.layer.1.layer_norm.weight model.encoder.block.1.layer.1.layer_norm.weight\n",
      "model.encoder.block.2.layer.0.SelfAttention.q.weight model.encoder.block.2.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.2.layer.0.SelfAttention.k.weight model.encoder.block.2.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.2.layer.0.SelfAttention.v.weight model.encoder.block.2.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.2.layer.0.SelfAttention.o.weight model.encoder.block.2.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.2.layer.0.layer_norm.weight model.encoder.block.2.layer.0.layer_norm.weight\n",
      "model.encoder.block.2.layer.1.DenseReluDense.wi_0.weight model.encoder.block.2.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.2.layer.1.DenseReluDense.wi_1.weight model.encoder.block.2.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.2.layer.1.DenseReluDense.wo.weight model.encoder.block.2.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.2.layer.1.layer_norm.weight model.encoder.block.2.layer.1.layer_norm.weight\n",
      "model.encoder.block.3.layer.0.SelfAttention.q.weight model.encoder.block.3.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.3.layer.0.SelfAttention.k.weight model.encoder.block.3.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.3.layer.0.SelfAttention.v.weight model.encoder.block.3.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.3.layer.0.SelfAttention.o.weight model.encoder.block.3.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.3.layer.0.layer_norm.weight model.encoder.block.3.layer.0.layer_norm.weight\n",
      "model.encoder.block.3.layer.1.DenseReluDense.wi_0.weight model.encoder.block.3.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.3.layer.1.DenseReluDense.wi_1.weight model.encoder.block.3.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.3.layer.1.DenseReluDense.wo.weight model.encoder.block.3.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.3.layer.1.layer_norm.weight model.encoder.block.3.layer.1.layer_norm.weight\n",
      "model.encoder.block.4.layer.0.SelfAttention.q.weight model.encoder.block.4.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.4.layer.0.SelfAttention.k.weight model.encoder.block.4.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.4.layer.0.SelfAttention.v.weight model.encoder.block.4.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.4.layer.0.SelfAttention.o.weight model.encoder.block.4.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.4.layer.0.layer_norm.weight model.encoder.block.4.layer.0.layer_norm.weight\n",
      "model.encoder.block.4.layer.1.DenseReluDense.wi_0.weight model.encoder.block.4.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.4.layer.1.DenseReluDense.wi_1.weight model.encoder.block.4.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.4.layer.1.DenseReluDense.wo.weight model.encoder.block.4.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.4.layer.1.layer_norm.weight model.encoder.block.4.layer.1.layer_norm.weight\n",
      "model.encoder.block.5.layer.0.SelfAttention.q.weight model.encoder.block.5.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.5.layer.0.SelfAttention.k.weight model.encoder.block.5.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.5.layer.0.SelfAttention.v.weight model.encoder.block.5.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.5.layer.0.SelfAttention.o.weight model.encoder.block.5.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.5.layer.0.layer_norm.weight model.encoder.block.5.layer.0.layer_norm.weight\n",
      "model.encoder.block.5.layer.1.DenseReluDense.wi_0.weight model.encoder.block.5.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.5.layer.1.DenseReluDense.wi_1.weight model.encoder.block.5.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.5.layer.1.DenseReluDense.wo.weight model.encoder.block.5.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.5.layer.1.layer_norm.weight model.encoder.block.5.layer.1.layer_norm.weight\n",
      "model.encoder.block.6.layer.0.SelfAttention.q.weight model.encoder.block.6.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.6.layer.0.SelfAttention.k.weight model.encoder.block.6.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.6.layer.0.SelfAttention.v.weight model.encoder.block.6.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.6.layer.0.SelfAttention.o.weight model.encoder.block.6.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.6.layer.0.layer_norm.weight model.encoder.block.6.layer.0.layer_norm.weight\n",
      "model.encoder.block.6.layer.1.DenseReluDense.wi_0.weight model.encoder.block.6.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.6.layer.1.DenseReluDense.wi_1.weight model.encoder.block.6.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.6.layer.1.DenseReluDense.wo.weight model.encoder.block.6.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.6.layer.1.layer_norm.weight model.encoder.block.6.layer.1.layer_norm.weight\n",
      "model.encoder.block.7.layer.0.SelfAttention.q.weight model.encoder.block.7.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.7.layer.0.SelfAttention.k.weight model.encoder.block.7.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.7.layer.0.SelfAttention.v.weight model.encoder.block.7.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.7.layer.0.SelfAttention.o.weight model.encoder.block.7.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.7.layer.0.layer_norm.weight model.encoder.block.7.layer.0.layer_norm.weight\n",
      "model.encoder.block.7.layer.1.DenseReluDense.wi_0.weight model.encoder.block.7.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.7.layer.1.DenseReluDense.wi_1.weight model.encoder.block.7.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.7.layer.1.DenseReluDense.wo.weight model.encoder.block.7.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.7.layer.1.layer_norm.weight model.encoder.block.7.layer.1.layer_norm.weight\n",
      "model.encoder.block.8.layer.0.SelfAttention.q.weight model.encoder.block.8.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.8.layer.0.SelfAttention.k.weight model.encoder.block.8.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.8.layer.0.SelfAttention.v.weight model.encoder.block.8.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.8.layer.0.SelfAttention.o.weight model.encoder.block.8.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.8.layer.0.layer_norm.weight model.encoder.block.8.layer.0.layer_norm.weight\n",
      "model.encoder.block.8.layer.1.DenseReluDense.wi_0.weight model.encoder.block.8.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.8.layer.1.DenseReluDense.wi_1.weight model.encoder.block.8.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.8.layer.1.DenseReluDense.wo.weight model.encoder.block.8.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.8.layer.1.layer_norm.weight model.encoder.block.8.layer.1.layer_norm.weight\n",
      "model.encoder.block.9.layer.0.SelfAttention.q.weight model.encoder.block.9.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.9.layer.0.SelfAttention.k.weight model.encoder.block.9.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.9.layer.0.SelfAttention.v.weight model.encoder.block.9.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.9.layer.0.SelfAttention.o.weight model.encoder.block.9.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.9.layer.0.layer_norm.weight model.encoder.block.9.layer.0.layer_norm.weight\n",
      "model.encoder.block.9.layer.1.DenseReluDense.wi_0.weight model.encoder.block.9.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.9.layer.1.DenseReluDense.wi_1.weight model.encoder.block.9.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.9.layer.1.DenseReluDense.wo.weight model.encoder.block.9.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.9.layer.1.layer_norm.weight model.encoder.block.9.layer.1.layer_norm.weight\n",
      "model.encoder.block.10.layer.0.SelfAttention.q.weight model.encoder.block.10.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.10.layer.0.SelfAttention.k.weight model.encoder.block.10.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.10.layer.0.SelfAttention.v.weight model.encoder.block.10.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.10.layer.0.SelfAttention.o.weight model.encoder.block.10.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.10.layer.0.layer_norm.weight model.encoder.block.10.layer.0.layer_norm.weight\n",
      "model.encoder.block.10.layer.1.DenseReluDense.wi_0.weight model.encoder.block.10.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.10.layer.1.DenseReluDense.wi_1.weight model.encoder.block.10.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.10.layer.1.DenseReluDense.wo.weight model.encoder.block.10.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.10.layer.1.layer_norm.weight model.encoder.block.10.layer.1.layer_norm.weight\n",
      "model.encoder.block.11.layer.0.SelfAttention.q.weight model.encoder.block.11.layer.0.SelfAttention.q.weight\n",
      "model.encoder.block.11.layer.0.SelfAttention.k.weight model.encoder.block.11.layer.0.SelfAttention.k.weight\n",
      "model.encoder.block.11.layer.0.SelfAttention.v.weight model.encoder.block.11.layer.0.SelfAttention.v.weight\n",
      "model.encoder.block.11.layer.0.SelfAttention.o.weight model.encoder.block.11.layer.0.SelfAttention.o.weight\n",
      "model.encoder.block.11.layer.0.layer_norm.weight model.encoder.block.11.layer.0.layer_norm.weight\n",
      "model.encoder.block.11.layer.1.DenseReluDense.wi_0.weight model.encoder.block.11.layer.1.DenseReluDense.wi_0.weight\n",
      "model.encoder.block.11.layer.1.DenseReluDense.wi_1.weight model.encoder.block.11.layer.1.DenseReluDense.wi_1.weight\n",
      "model.encoder.block.11.layer.1.DenseReluDense.wo.weight model.encoder.block.11.layer.1.DenseReluDense.wo.weight\n",
      "model.encoder.block.11.layer.1.layer_norm.weight model.encoder.block.11.layer.1.layer_norm.weight\n",
      "model.encoder.final_layer_norm.weight model.encoder.final_layer_norm.weight\n",
      "model.decoder.embed_tokens.weight model.decoder.embed_tokens.weight\n",
      "model.decoder.block.0.layer.0.SelfAttention.q.weight model.decoder.block.0.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.0.layer.0.SelfAttention.k.weight model.decoder.block.0.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.0.layer.0.SelfAttention.v.weight model.decoder.block.0.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.0.layer.0.SelfAttention.o.weight model.decoder.block.0.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.0.layer.0.SelfAttention.relative_attention_bias.weight model.decoder.block.0.layer.0.SelfAttention.relative_attention_bias.weight\n",
      "model.decoder.block.0.layer.0.layer_norm.weight model.decoder.block.0.layer.0.layer_norm.weight\n",
      "model.decoder.block.0.layer.1.EncDecAttention.q.weight model.decoder.block.0.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.0.layer.1.EncDecAttention.k.weight model.decoder.block.0.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.0.layer.1.EncDecAttention.v.weight model.decoder.block.0.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.0.layer.1.EncDecAttention.o.weight model.decoder.block.0.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.0.layer.1.layer_norm.weight model.decoder.block.0.layer.1.layer_norm.weight\n",
      "model.decoder.block.0.layer.2.DenseReluDense.wi_0.weight model.decoder.block.0.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.0.layer.2.DenseReluDense.wi_1.weight model.decoder.block.0.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.0.layer.2.DenseReluDense.wo.weight model.decoder.block.0.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.0.layer.2.layer_norm.weight model.decoder.block.0.layer.2.layer_norm.weight\n",
      "model.decoder.block.1.layer.0.SelfAttention.q.weight model.decoder.block.1.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.1.layer.0.SelfAttention.k.weight model.decoder.block.1.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.1.layer.0.SelfAttention.v.weight model.decoder.block.1.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.1.layer.0.SelfAttention.o.weight model.decoder.block.1.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.1.layer.0.layer_norm.weight model.decoder.block.1.layer.0.layer_norm.weight\n",
      "model.decoder.block.1.layer.1.EncDecAttention.q.weight model.decoder.block.1.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.1.layer.1.EncDecAttention.k.weight model.decoder.block.1.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.1.layer.1.EncDecAttention.v.weight model.decoder.block.1.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.1.layer.1.EncDecAttention.o.weight model.decoder.block.1.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.1.layer.1.layer_norm.weight model.decoder.block.1.layer.1.layer_norm.weight\n",
      "model.decoder.block.1.layer.2.DenseReluDense.wi_0.weight model.decoder.block.1.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.1.layer.2.DenseReluDense.wi_1.weight model.decoder.block.1.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.1.layer.2.DenseReluDense.wo.weight model.decoder.block.1.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.1.layer.2.layer_norm.weight model.decoder.block.1.layer.2.layer_norm.weight\n",
      "model.decoder.block.2.layer.0.SelfAttention.q.weight model.decoder.block.2.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.2.layer.0.SelfAttention.k.weight model.decoder.block.2.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.2.layer.0.SelfAttention.v.weight model.decoder.block.2.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.2.layer.0.SelfAttention.o.weight model.decoder.block.2.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.2.layer.0.layer_norm.weight model.decoder.block.2.layer.0.layer_norm.weight\n",
      "model.decoder.block.2.layer.1.EncDecAttention.q.weight model.decoder.block.2.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.2.layer.1.EncDecAttention.k.weight model.decoder.block.2.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.2.layer.1.EncDecAttention.v.weight model.decoder.block.2.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.2.layer.1.EncDecAttention.o.weight model.decoder.block.2.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.2.layer.1.layer_norm.weight model.decoder.block.2.layer.1.layer_norm.weight\n",
      "model.decoder.block.2.layer.2.DenseReluDense.wi_0.weight model.decoder.block.2.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.2.layer.2.DenseReluDense.wi_1.weight model.decoder.block.2.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.2.layer.2.DenseReluDense.wo.weight model.decoder.block.2.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.2.layer.2.layer_norm.weight model.decoder.block.2.layer.2.layer_norm.weight\n",
      "model.decoder.block.3.layer.0.SelfAttention.q.weight model.decoder.block.3.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.3.layer.0.SelfAttention.k.weight model.decoder.block.3.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.3.layer.0.SelfAttention.v.weight model.decoder.block.3.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.3.layer.0.SelfAttention.o.weight model.decoder.block.3.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.3.layer.0.layer_norm.weight model.decoder.block.3.layer.0.layer_norm.weight\n",
      "model.decoder.block.3.layer.1.EncDecAttention.q.weight model.decoder.block.3.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.3.layer.1.EncDecAttention.k.weight model.decoder.block.3.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.3.layer.1.EncDecAttention.v.weight model.decoder.block.3.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.3.layer.1.EncDecAttention.o.weight model.decoder.block.3.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.3.layer.1.layer_norm.weight model.decoder.block.3.layer.1.layer_norm.weight\n",
      "model.decoder.block.3.layer.2.DenseReluDense.wi_0.weight model.decoder.block.3.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.3.layer.2.DenseReluDense.wi_1.weight model.decoder.block.3.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.3.layer.2.DenseReluDense.wo.weight model.decoder.block.3.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.3.layer.2.layer_norm.weight model.decoder.block.3.layer.2.layer_norm.weight\n",
      "model.decoder.block.4.layer.0.SelfAttention.q.weight model.decoder.block.4.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.4.layer.0.SelfAttention.k.weight model.decoder.block.4.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.4.layer.0.SelfAttention.v.weight model.decoder.block.4.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.4.layer.0.SelfAttention.o.weight model.decoder.block.4.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.4.layer.0.layer_norm.weight model.decoder.block.4.layer.0.layer_norm.weight\n",
      "model.decoder.block.4.layer.1.EncDecAttention.q.weight model.decoder.block.4.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.4.layer.1.EncDecAttention.k.weight model.decoder.block.4.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.4.layer.1.EncDecAttention.v.weight model.decoder.block.4.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.4.layer.1.EncDecAttention.o.weight model.decoder.block.4.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.4.layer.1.layer_norm.weight model.decoder.block.4.layer.1.layer_norm.weight\n",
      "model.decoder.block.4.layer.2.DenseReluDense.wi_0.weight model.decoder.block.4.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.4.layer.2.DenseReluDense.wi_1.weight model.decoder.block.4.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.4.layer.2.DenseReluDense.wo.weight model.decoder.block.4.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.4.layer.2.layer_norm.weight model.decoder.block.4.layer.2.layer_norm.weight\n",
      "model.decoder.block.5.layer.0.SelfAttention.q.weight model.decoder.block.5.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.5.layer.0.SelfAttention.k.weight model.decoder.block.5.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.5.layer.0.SelfAttention.v.weight model.decoder.block.5.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.5.layer.0.SelfAttention.o.weight model.decoder.block.5.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.5.layer.0.layer_norm.weight model.decoder.block.5.layer.0.layer_norm.weight\n",
      "model.decoder.block.5.layer.1.EncDecAttention.q.weight model.decoder.block.5.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.5.layer.1.EncDecAttention.k.weight model.decoder.block.5.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.5.layer.1.EncDecAttention.v.weight model.decoder.block.5.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.5.layer.1.EncDecAttention.o.weight model.decoder.block.5.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.5.layer.1.layer_norm.weight model.decoder.block.5.layer.1.layer_norm.weight\n",
      "model.decoder.block.5.layer.2.DenseReluDense.wi_0.weight model.decoder.block.5.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.5.layer.2.DenseReluDense.wi_1.weight model.decoder.block.5.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.5.layer.2.DenseReluDense.wo.weight model.decoder.block.5.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.5.layer.2.layer_norm.weight model.decoder.block.5.layer.2.layer_norm.weight\n",
      "model.decoder.block.6.layer.0.SelfAttention.q.weight model.decoder.block.6.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.6.layer.0.SelfAttention.k.weight model.decoder.block.6.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.6.layer.0.SelfAttention.v.weight model.decoder.block.6.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.6.layer.0.SelfAttention.o.weight model.decoder.block.6.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.6.layer.0.layer_norm.weight model.decoder.block.6.layer.0.layer_norm.weight\n",
      "model.decoder.block.6.layer.1.EncDecAttention.q.weight model.decoder.block.6.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.6.layer.1.EncDecAttention.k.weight model.decoder.block.6.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.6.layer.1.EncDecAttention.v.weight model.decoder.block.6.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.6.layer.1.EncDecAttention.o.weight model.decoder.block.6.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.6.layer.1.layer_norm.weight model.decoder.block.6.layer.1.layer_norm.weight\n",
      "model.decoder.block.6.layer.2.DenseReluDense.wi_0.weight model.decoder.block.6.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.6.layer.2.DenseReluDense.wi_1.weight model.decoder.block.6.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.6.layer.2.DenseReluDense.wo.weight model.decoder.block.6.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.6.layer.2.layer_norm.weight model.decoder.block.6.layer.2.layer_norm.weight\n",
      "model.decoder.block.7.layer.0.SelfAttention.q.weight model.decoder.block.7.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.7.layer.0.SelfAttention.k.weight model.decoder.block.7.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.7.layer.0.SelfAttention.v.weight model.decoder.block.7.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.7.layer.0.SelfAttention.o.weight model.decoder.block.7.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.7.layer.0.layer_norm.weight model.decoder.block.7.layer.0.layer_norm.weight\n",
      "model.decoder.block.7.layer.1.EncDecAttention.q.weight model.decoder.block.7.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.7.layer.1.EncDecAttention.k.weight model.decoder.block.7.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.7.layer.1.EncDecAttention.v.weight model.decoder.block.7.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.7.layer.1.EncDecAttention.o.weight model.decoder.block.7.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.7.layer.1.layer_norm.weight model.decoder.block.7.layer.1.layer_norm.weight\n",
      "model.decoder.block.7.layer.2.DenseReluDense.wi_0.weight model.decoder.block.7.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.7.layer.2.DenseReluDense.wi_1.weight model.decoder.block.7.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.7.layer.2.DenseReluDense.wo.weight model.decoder.block.7.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.7.layer.2.layer_norm.weight model.decoder.block.7.layer.2.layer_norm.weight\n",
      "model.decoder.block.8.layer.0.SelfAttention.q.weight model.decoder.block.8.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.8.layer.0.SelfAttention.k.weight model.decoder.block.8.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.8.layer.0.SelfAttention.v.weight model.decoder.block.8.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.8.layer.0.SelfAttention.o.weight model.decoder.block.8.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.8.layer.0.layer_norm.weight model.decoder.block.8.layer.0.layer_norm.weight\n",
      "model.decoder.block.8.layer.1.EncDecAttention.q.weight model.decoder.block.8.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.8.layer.1.EncDecAttention.k.weight model.decoder.block.8.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.8.layer.1.EncDecAttention.v.weight model.decoder.block.8.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.8.layer.1.EncDecAttention.o.weight model.decoder.block.8.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.8.layer.1.layer_norm.weight model.decoder.block.8.layer.1.layer_norm.weight\n",
      "model.decoder.block.8.layer.2.DenseReluDense.wi_0.weight model.decoder.block.8.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.8.layer.2.DenseReluDense.wi_1.weight model.decoder.block.8.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.8.layer.2.DenseReluDense.wo.weight model.decoder.block.8.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.8.layer.2.layer_norm.weight model.decoder.block.8.layer.2.layer_norm.weight\n",
      "model.decoder.block.9.layer.0.SelfAttention.q.weight model.decoder.block.9.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.9.layer.0.SelfAttention.k.weight model.decoder.block.9.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.9.layer.0.SelfAttention.v.weight model.decoder.block.9.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.9.layer.0.SelfAttention.o.weight model.decoder.block.9.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.9.layer.0.layer_norm.weight model.decoder.block.9.layer.0.layer_norm.weight\n",
      "model.decoder.block.9.layer.1.EncDecAttention.q.weight model.decoder.block.9.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.9.layer.1.EncDecAttention.k.weight model.decoder.block.9.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.9.layer.1.EncDecAttention.v.weight model.decoder.block.9.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.9.layer.1.EncDecAttention.o.weight model.decoder.block.9.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.9.layer.1.layer_norm.weight model.decoder.block.9.layer.1.layer_norm.weight\n",
      "model.decoder.block.9.layer.2.DenseReluDense.wi_0.weight model.decoder.block.9.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.9.layer.2.DenseReluDense.wi_1.weight model.decoder.block.9.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.9.layer.2.DenseReluDense.wo.weight model.decoder.block.9.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.9.layer.2.layer_norm.weight model.decoder.block.9.layer.2.layer_norm.weight\n",
      "model.decoder.block.10.layer.0.SelfAttention.q.weight model.decoder.block.10.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.10.layer.0.SelfAttention.k.weight model.decoder.block.10.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.10.layer.0.SelfAttention.v.weight model.decoder.block.10.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.10.layer.0.SelfAttention.o.weight model.decoder.block.10.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.10.layer.0.layer_norm.weight model.decoder.block.10.layer.0.layer_norm.weight\n",
      "model.decoder.block.10.layer.1.EncDecAttention.q.weight model.decoder.block.10.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.10.layer.1.EncDecAttention.k.weight model.decoder.block.10.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.10.layer.1.EncDecAttention.v.weight model.decoder.block.10.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.10.layer.1.EncDecAttention.o.weight model.decoder.block.10.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.10.layer.1.layer_norm.weight model.decoder.block.10.layer.1.layer_norm.weight\n",
      "model.decoder.block.10.layer.2.DenseReluDense.wi_0.weight model.decoder.block.10.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.10.layer.2.DenseReluDense.wi_1.weight model.decoder.block.10.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.10.layer.2.DenseReluDense.wo.weight model.decoder.block.10.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.10.layer.2.layer_norm.weight model.decoder.block.10.layer.2.layer_norm.weight\n",
      "model.decoder.block.11.layer.0.SelfAttention.q.weight model.decoder.block.11.layer.0.SelfAttention.q.weight\n",
      "model.decoder.block.11.layer.0.SelfAttention.k.weight model.decoder.block.11.layer.0.SelfAttention.k.weight\n",
      "model.decoder.block.11.layer.0.SelfAttention.v.weight model.decoder.block.11.layer.0.SelfAttention.v.weight\n",
      "model.decoder.block.11.layer.0.SelfAttention.o.weight model.decoder.block.11.layer.0.SelfAttention.o.weight\n",
      "model.decoder.block.11.layer.0.layer_norm.weight model.decoder.block.11.layer.0.layer_norm.weight\n",
      "model.decoder.block.11.layer.1.EncDecAttention.q.weight model.decoder.block.11.layer.1.EncDecAttention.q.weight\n",
      "model.decoder.block.11.layer.1.EncDecAttention.k.weight model.decoder.block.11.layer.1.EncDecAttention.k.weight\n",
      "model.decoder.block.11.layer.1.EncDecAttention.v.weight model.decoder.block.11.layer.1.EncDecAttention.v.weight\n",
      "model.decoder.block.11.layer.1.EncDecAttention.o.weight model.decoder.block.11.layer.1.EncDecAttention.o.weight\n",
      "model.decoder.block.11.layer.1.layer_norm.weight model.decoder.block.11.layer.1.layer_norm.weight\n",
      "model.decoder.block.11.layer.2.DenseReluDense.wi_0.weight model.decoder.block.11.layer.2.DenseReluDense.wi_0.weight\n",
      "model.decoder.block.11.layer.2.DenseReluDense.wi_1.weight model.decoder.block.11.layer.2.DenseReluDense.wi_1.weight\n",
      "model.decoder.block.11.layer.2.DenseReluDense.wo.weight model.decoder.block.11.layer.2.DenseReluDense.wo.weight\n",
      "model.decoder.block.11.layer.2.layer_norm.weight model.decoder.block.11.layer.2.layer_norm.weight\n",
      "model.decoder.final_layer_norm.weight model.decoder.final_layer_norm.weight\n",
      "model.lm_head.weight model.lm_head.weight\n"
     ]
    }
   ],
   "source": [
    "for k, v in old_weights:\n",
    "    new_k = k.replace('._orig_mod', '')\n",
    "    print(k, new_k)\n",
    "    weights[new_k] = v"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<All keys matched successfully>"
      ]
     },
     "execution_count": 36,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "model.load_state_dict(weights)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "Seq2SeqLMOutput(loss=tensor(3.0237, grad_fn=<NllLossBackward0>), logits=tensor([[[-0.9907,  2.8476,  1.9752,  ..., -2.8603, -2.6576, -1.2661],\n",
       "         [-1.8009,  3.8409,  3.7604,  ..., -4.5560, -2.4641, -3.2539],\n",
       "         [-1.6907,  0.0464,  7.7736,  ..., -2.7157, -4.4345, -3.7622],\n",
       "         ...,\n",
       "         [-1.6395,  1.4647,  4.4011,  ..., -2.5878, -0.5852, -3.5563],\n",
       "         [-2.6389,  2.4625,  5.2757,  ..., -6.3230, -4.1409, -2.0036],\n",
       "         [ 0.3584,  2.3890, 35.3724,  ..., -1.7918, -3.8099, -0.6635]],\n",
       "\n",
       "        [[-0.6767,  2.5914,  2.1937,  ..., -2.3515, -2.3793, -1.3622],\n",
       "         [-4.3737,  5.6130,  4.0856,  ..., -2.0719, -2.5653, -0.1653],\n",
       "         [-3.6748, -0.1740,  2.9097,  ..., -2.9931, -3.7182, -3.8249],\n",
       "         ...,\n",
       "         [-1.5893,  0.0879,  4.5392,  ..., -2.3501, -1.0870, -5.1463],\n",
       "         [-3.1135,  0.4900,  6.3222,  ..., -3.0063, -1.8968, -4.1890],\n",
       "         [-0.3985,  2.4653, 27.7797,  ..., -2.1855, -2.1259, -2.1988]],\n",
       "\n",
       "        [[-0.6180,  2.8315,  2.3604,  ..., -2.5402, -2.5089, -1.1249],\n",
       "         [-2.4090,  3.0879,  6.2518,  ..., -3.7311, -3.0210, -2.8807],\n",
       "         [-3.0661,  2.6362,  6.5863,  ..., -3.7160, -3.0774, -2.6890],\n",
       "         ...,\n",
       "         [-1.2520,  1.1135,  6.9891,  ..., -4.3358, -3.3024, -3.8073],\n",
       "         [-0.6629,  3.7204,  9.6339,  ..., -3.8994, -2.9204, -0.7210],\n",
       "         [ 0.1121,  3.1163, 35.7059,  ..., -1.5697, -3.0881, -3.1882]]],\n",
       "       grad_fn=<UnsafeViewBackward0>), past_key_values=((tensor([[[[-2.9489e-01, -1.1060e+00, -8.1082e-01,  ...,  4.4001e-01,\n",
       "            1.4864e+00, -6.0747e-01],\n",
       "          [-4.6619e+00, -6.1707e+00,  7.6104e-01,  ..., -2.9147e+00,\n",
       "            8.2311e-01, -1.0113e-01],\n",
       "          [-4.1654e-02,  6.6000e-01, -3.4746e-01,  ..., -6.3220e-01,\n",
       "           -1.9728e-01,  6.5129e-02],\n",
       "          ...,\n",
       "          [ 2.3770e-01,  1.2191e-01,  2.9408e-01,  ..., -3.8956e-02,\n",
       "           -3.2108e-01, -8.9294e-02],\n",
       "          [ 2.0296e-01, -1.2188e+00,  1.1838e+00,  ..., -5.8493e-02,\n",
       "            2.4833e-01,  3.8312e-01],\n",
       "          [-3.8785e-01, -5.9471e-02, -7.8156e-01,  ..., -5.3697e-02,\n",
       "            6.1625e-01,  1.2986e-01]],\n",
       "\n",
       "         [[-2.2115e+00, -1.2244e+00, -1.3060e+00,  ..., -2.0759e+00,\n",
       "           -8.8623e-01,  2.6129e+00],\n",
       "          [-2.7877e+00, -1.4573e-01,  1.7794e+00,  ..., -3.2485e-01,\n",
       "            9.5459e-01,  1.1867e+00],\n",
       "          [ 2.4029e-01,  1.1682e-01,  4.8660e-01,  ...,  3.8699e-01,\n",
       "            4.3397e-01,  5.0144e-01],\n",
       "          ...,\n",
       "          [ 5.8826e-01, -1.7348e-02,  3.1557e-01,  ...,  2.0696e-01,\n",
       "            5.4701e-01, -4.3229e-01],\n",
       "          [-3.4117e-01,  4.8919e-01, -1.8541e-01,  ...,  5.5052e-02,\n",
       "            9.2583e-01, -3.7684e-01],\n",
       "          [-4.9976e-01,  2.2597e-01,  3.0643e-01,  ...,  1.6291e-02,\n",
       "            5.5030e-01, -5.5294e-01]],\n",
       "\n",
       "         [[-8.0358e-01, -2.0435e+00, -2.1926e+00,  ..., -1.6051e+00,\n",
       "            3.1620e+00, -2.9805e+00],\n",
       "          [-3.7822e+00,  1.5057e+00,  1.0497e+00,  ...,  3.0560e+00,\n",
       "           -1.5511e+00, -7.9590e-01],\n",
       "          [ 1.6567e-01, -6.3222e-01,  5.6686e-02,  ...,  1.0332e-01,\n",
       "           -1.2141e-01, -6.2444e-02],\n",
       "          ...,\n",
       "          [ 2.1651e-01, -4.3542e-01,  5.7137e-02,  ..., -5.1990e-01,\n",
       "            1.0585e-01, -4.4621e-01],\n",
       "          [-8.6824e-01, -4.3131e-01,  1.8410e-01,  ...,  1.4391e-01,\n",
       "           -8.9246e-01,  5.5248e-01],\n",
       "          [-1.6334e-02, -2.1831e-01,  1.2294e-01,  ..., -3.5481e-01,\n",
       "           -6.8424e-01, -3.8839e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.8533e-01,  1.9913e+00,  2.0568e+00,  ...,  2.8268e-01,\n",
       "            1.5363e+00,  3.8949e+00],\n",
       "          [-2.8026e+00,  3.2078e+00,  1.7583e+00,  ...,  1.8930e+00,\n",
       "            3.7360e+00,  5.0645e-01],\n",
       "          [-2.5199e-01,  1.0396e+00,  9.4241e-01,  ...,  8.4198e-02,\n",
       "            6.7324e-01, -1.4692e-01],\n",
       "          ...,\n",
       "          [-4.9355e-01, -3.5045e-01, -1.9307e-02,  ...,  2.5524e-01,\n",
       "            1.1853e-01,  1.1316e+00],\n",
       "          [-3.5792e-01,  6.2817e-01,  3.3269e-01,  ...,  1.6622e-01,\n",
       "            8.1888e-03,  2.0611e-01],\n",
       "          [ 2.4341e-01,  1.2003e-01, -5.6291e-02,  ...,  5.0918e-01,\n",
       "            1.7581e-01, -5.2497e-01]],\n",
       "\n",
       "         [[-2.5090e+00, -2.4575e+00,  3.6387e-01,  ..., -1.9372e+00,\n",
       "           -2.5834e+00, -2.6079e+00],\n",
       "          [-3.1768e+00, -6.9764e-02,  3.3799e+00,  ...,  1.4937e+00,\n",
       "            1.5066e+00,  1.9805e+00],\n",
       "          [-4.7495e-01, -1.3115e-01,  2.4587e-02,  ...,  3.3698e-01,\n",
       "           -3.7733e-02,  3.6014e-01],\n",
       "          ...,\n",
       "          [-3.6233e-01, -8.5387e-01, -1.0281e+00,  ..., -2.3507e-02,\n",
       "            2.1315e-01, -1.0669e+00],\n",
       "          [-3.9683e-01, -1.1791e+00,  6.4318e-01,  ..., -5.3635e-01,\n",
       "            5.3644e-02, -9.6502e-01],\n",
       "          [-4.1113e-01, -4.4001e-01,  5.8775e-02,  ..., -6.6012e-01,\n",
       "           -2.9521e-01, -8.5738e-01]],\n",
       "\n",
       "         [[-1.7151e-01,  9.0868e-01, -3.5472e+00,  ...,  3.6642e+00,\n",
       "            1.7292e+00,  3.1294e+00],\n",
       "          [-1.1900e+00, -2.2806e-01,  1.6730e+00,  ..., -3.1227e+00,\n",
       "           -1.9874e+00, -2.8181e+00],\n",
       "          [-6.3067e-01,  1.8235e-01,  1.5574e-01,  ...,  8.4381e-01,\n",
       "            1.2978e-01,  1.0430e+00],\n",
       "          ...,\n",
       "          [ 1.1410e-01, -2.5354e-01,  1.5042e-01,  ...,  4.8572e-01,\n",
       "           -3.9025e-01, -1.3540e-01],\n",
       "          [-3.7126e-01, -5.8723e-01,  1.5729e+00,  ..., -1.7447e+00,\n",
       "           -9.6563e-01, -2.8337e-01],\n",
       "          [-1.7131e-01,  3.4864e-01,  2.5864e-01,  ..., -4.4751e-01,\n",
       "            6.4733e-01,  2.5766e-02]]],\n",
       "\n",
       "\n",
       "        [[[-2.9489e-01, -1.1060e+00, -8.1082e-01,  ...,  4.4001e-01,\n",
       "            1.4864e+00, -6.0747e-01],\n",
       "          [-4.6619e+00, -6.1707e+00,  7.6104e-01,  ..., -2.9147e+00,\n",
       "            8.2311e-01, -1.0113e-01],\n",
       "          [-7.4037e-01, -1.7047e-01,  3.4783e-01,  ..., -1.3567e-01,\n",
       "            4.8516e-01, -4.2624e-02],\n",
       "          ...,\n",
       "          [-4.2473e-01,  2.4243e-01,  4.6504e-01,  ..., -7.9573e-01,\n",
       "            5.8055e-01,  6.0925e-02],\n",
       "          [ 2.0296e-01, -1.2188e+00,  1.1838e+00,  ..., -5.8493e-02,\n",
       "            2.4833e-01,  3.8312e-01],\n",
       "          [-5.6667e-01,  1.8782e-01,  1.3713e-01,  ...,  4.3147e-01,\n",
       "           -4.0919e-01, -3.2029e-01]],\n",
       "\n",
       "         [[-2.2115e+00, -1.2244e+00, -1.3060e+00,  ..., -2.0759e+00,\n",
       "           -8.8623e-01,  2.6129e+00],\n",
       "          [-2.7877e+00, -1.4573e-01,  1.7794e+00,  ..., -3.2485e-01,\n",
       "            9.5459e-01,  1.1867e+00],\n",
       "          [ 5.8584e-02, -4.3559e-01,  7.9758e-01,  ..., -1.4922e-01,\n",
       "            1.7614e-01,  2.5652e-01],\n",
       "          ...,\n",
       "          [ 3.7748e-01,  1.0586e+00,  2.7251e-01,  ..., -8.0130e-01,\n",
       "            2.1980e-01,  1.0732e-01],\n",
       "          [-3.4117e-01,  4.8919e-01, -1.8541e-01,  ...,  5.5052e-02,\n",
       "            9.2583e-01, -3.7684e-01],\n",
       "          [ 2.1983e-01,  7.4621e-01, -1.6219e+00,  ...,  2.0996e-01,\n",
       "           -1.3473e-01,  2.2753e-01]],\n",
       "\n",
       "         [[-8.0358e-01, -2.0435e+00, -2.1926e+00,  ..., -1.6051e+00,\n",
       "            3.1620e+00, -2.9805e+00],\n",
       "          [-3.7822e+00,  1.5057e+00,  1.0497e+00,  ...,  3.0560e+00,\n",
       "           -1.5511e+00, -7.9590e-01],\n",
       "          [-7.4475e-01,  6.3389e-01,  3.7253e-01,  ...,  4.8596e-01,\n",
       "            4.0336e-03,  6.9843e-02],\n",
       "          ...,\n",
       "          [ 4.4012e-01,  3.6837e-01, -4.6814e-01,  ..., -7.7475e-01,\n",
       "            5.0105e-01, -3.4946e-01],\n",
       "          [-8.6824e-01, -4.3131e-01,  1.8410e-01,  ...,  1.4391e-01,\n",
       "           -8.9246e-01,  5.5248e-01],\n",
       "          [ 6.0972e-01,  4.3153e-01, -2.4905e-01,  ...,  1.8612e-01,\n",
       "            4.1217e-02,  1.9362e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.8533e-01,  1.9913e+00,  2.0568e+00,  ...,  2.8268e-01,\n",
       "            1.5363e+00,  3.8949e+00],\n",
       "          [-2.8026e+00,  3.2078e+00,  1.7583e+00,  ...,  1.8930e+00,\n",
       "            3.7360e+00,  5.0645e-01],\n",
       "          [-4.4550e-01, -1.3672e-01, -3.2727e-01,  ..., -4.1879e-01,\n",
       "            1.5698e-01, -7.0899e-01],\n",
       "          ...,\n",
       "          [-6.0894e-01,  2.8779e-01,  6.3227e-01,  ..., -5.2507e-02,\n",
       "            3.3380e-01, -8.7367e-02],\n",
       "          [-3.5792e-01,  6.2817e-01,  3.3269e-01,  ...,  1.6622e-01,\n",
       "            8.1888e-03,  2.0611e-01],\n",
       "          [ 4.6539e-01, -5.8827e-02, -5.3714e-01,  ..., -7.0437e-02,\n",
       "           -4.1367e-01,  9.6030e-01]],\n",
       "\n",
       "         [[-2.5090e+00, -2.4575e+00,  3.6387e-01,  ..., -1.9372e+00,\n",
       "           -2.5834e+00, -2.6079e+00],\n",
       "          [-3.1768e+00, -6.9764e-02,  3.3799e+00,  ...,  1.4937e+00,\n",
       "            1.5066e+00,  1.9805e+00],\n",
       "          [-2.7708e-01, -1.3633e-01,  1.9732e-01,  ...,  2.4425e-02,\n",
       "            4.9351e-01,  6.0038e-01],\n",
       "          ...,\n",
       "          [ 3.6723e-01, -6.6187e-01, -9.3616e-02,  ...,  2.9338e-02,\n",
       "            9.8299e-01,  5.0122e-01],\n",
       "          [-3.9683e-01, -1.1791e+00,  6.4318e-01,  ..., -5.3635e-01,\n",
       "            5.3644e-02, -9.6502e-01],\n",
       "          [ 7.4586e-01,  1.0313e+00, -4.3902e-01,  ..., -5.1329e-01,\n",
       "            8.2926e-02, -6.0176e-01]],\n",
       "\n",
       "         [[-1.7151e-01,  9.0868e-01, -3.5472e+00,  ...,  3.6642e+00,\n",
       "            1.7292e+00,  3.1294e+00],\n",
       "          [-1.1900e+00, -2.2806e-01,  1.6730e+00,  ..., -3.1227e+00,\n",
       "           -1.9874e+00, -2.8181e+00],\n",
       "          [ 1.6827e-01,  1.0741e-01,  3.6662e-01,  ..., -1.6843e+00,\n",
       "           -4.3744e-01, -6.2487e-01],\n",
       "          ...,\n",
       "          [-4.6486e-01,  5.4911e-01, -5.1425e-01,  ...,  7.7752e-01,\n",
       "           -3.2402e-02,  5.4745e-01],\n",
       "          [-3.7126e-01, -5.8723e-01,  1.5729e+00,  ..., -1.7447e+00,\n",
       "           -9.6563e-01, -2.8337e-01],\n",
       "          [-1.2330e-01,  3.0027e-01,  9.1729e-01,  ...,  1.9355e-01,\n",
       "           -2.9699e-01,  3.3482e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.9489e-01, -1.1060e+00, -8.1082e-01,  ...,  4.4001e-01,\n",
       "            1.4864e+00, -6.0747e-01],\n",
       "          [-4.6619e+00, -6.1707e+00,  7.6104e-01,  ..., -2.9147e+00,\n",
       "            8.2311e-01, -1.0113e-01],\n",
       "          [-3.4278e-01, -5.4616e-01, -1.4316e+00,  ...,  8.9183e-01,\n",
       "            8.0808e-01,  5.1960e-01],\n",
       "          ...,\n",
       "          [-3.5413e-02, -2.1662e-01, -1.0494e-01,  ...,  5.2166e-01,\n",
       "            6.4178e-01,  8.5020e-01],\n",
       "          [-5.0940e-01, -1.5454e+00,  3.1565e-01,  ..., -1.3425e-01,\n",
       "            1.5078e+00, -7.6692e-01],\n",
       "          [ 5.5336e-01, -3.5436e-01,  9.9677e-03,  ...,  2.6348e-01,\n",
       "            5.1948e-01,  6.3569e-02]],\n",
       "\n",
       "         [[-2.2115e+00, -1.2244e+00, -1.3060e+00,  ..., -2.0759e+00,\n",
       "           -8.8623e-01,  2.6129e+00],\n",
       "          [-2.7877e+00, -1.4573e-01,  1.7794e+00,  ..., -3.2485e-01,\n",
       "            9.5459e-01,  1.1867e+00],\n",
       "          [-7.4429e-01,  9.7585e-01, -4.6837e-01,  ..., -3.3383e-02,\n",
       "           -9.2480e-02,  6.5081e-01],\n",
       "          ...,\n",
       "          [-6.6320e-01,  4.2758e-01, -4.1745e-01,  ...,  1.7848e-02,\n",
       "           -5.6702e-01, -1.7307e-01],\n",
       "          [ 7.2226e-01,  4.6044e-01, -7.8412e-01,  ..., -7.5874e-01,\n",
       "           -7.1358e-02, -3.4132e-01],\n",
       "          [-2.6542e-01, -7.3879e-01,  9.1938e-01,  ...,  5.4545e-01,\n",
       "            5.6048e-01,  2.3500e-01]],\n",
       "\n",
       "         [[-8.0358e-01, -2.0435e+00, -2.1926e+00,  ..., -1.6051e+00,\n",
       "            3.1620e+00, -2.9805e+00],\n",
       "          [-3.7822e+00,  1.5057e+00,  1.0497e+00,  ...,  3.0560e+00,\n",
       "           -1.5511e+00, -7.9590e-01],\n",
       "          [-9.7551e-01,  1.8396e-01,  4.5813e-01,  ...,  3.9379e-01,\n",
       "           -2.7435e+00, -2.0516e+00],\n",
       "          ...,\n",
       "          [ 5.2216e-01, -7.3102e-01, -5.2877e-01,  ..., -4.5898e-01,\n",
       "           -4.5983e-01, -5.7910e-01],\n",
       "          [-1.2057e+00,  8.1031e-01, -3.7878e-01,  ...,  6.4033e-01,\n",
       "           -1.6664e+00, -1.8176e+00],\n",
       "          [-1.1926e-01, -1.6181e-01, -5.9085e-01,  ..., -2.0915e-02,\n",
       "           -2.2606e-01,  1.1043e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.8533e-01,  1.9913e+00,  2.0568e+00,  ...,  2.8268e-01,\n",
       "            1.5363e+00,  3.8949e+00],\n",
       "          [-2.8026e+00,  3.2078e+00,  1.7583e+00,  ...,  1.8930e+00,\n",
       "            3.7360e+00,  5.0645e-01],\n",
       "          [-2.6081e-01,  7.8129e-01, -1.5962e-01,  ...,  3.9026e-01,\n",
       "            3.3083e-01, -5.5628e-01],\n",
       "          ...,\n",
       "          [ 6.5882e-02, -1.0552e-01,  1.0438e-01,  ..., -1.4406e-01,\n",
       "           -6.4327e-01,  3.3898e-01],\n",
       "          [-5.8450e-01,  2.5168e-01,  5.8061e-01,  ...,  5.4447e-01,\n",
       "           -1.7871e-02, -6.5251e-01],\n",
       "          [ 1.2958e-01, -1.2351e-01, -3.5363e-01,  ..., -1.0852e-02,\n",
       "           -2.5546e-01, -2.1535e-01]],\n",
       "\n",
       "         [[-2.5090e+00, -2.4575e+00,  3.6387e-01,  ..., -1.9372e+00,\n",
       "           -2.5834e+00, -2.6079e+00],\n",
       "          [-3.1768e+00, -6.9764e-02,  3.3799e+00,  ...,  1.4937e+00,\n",
       "            1.5066e+00,  1.9805e+00],\n",
       "          [-5.5444e-01,  9.3792e-01,  1.2124e+00,  ...,  8.9135e-01,\n",
       "            8.1035e-01,  1.0775e-01],\n",
       "          ...,\n",
       "          [ 6.2550e-01,  6.9921e-01,  9.9288e-01,  ...,  7.3010e-01,\n",
       "            9.6034e-01,  5.1372e-01],\n",
       "          [-1.4466e+00, -3.6540e-02,  2.1137e-01,  ...,  8.5881e-01,\n",
       "            1.4419e+00, -8.6848e-01],\n",
       "          [ 2.3099e-01, -4.3919e-01, -2.8730e-01,  ...,  1.2183e-01,\n",
       "            5.1394e-01, -4.9714e-01]],\n",
       "\n",
       "         [[-1.7151e-01,  9.0868e-01, -3.5472e+00,  ...,  3.6642e+00,\n",
       "            1.7292e+00,  3.1294e+00],\n",
       "          [-1.1900e+00, -2.2806e-01,  1.6730e+00,  ..., -3.1227e+00,\n",
       "           -1.9874e+00, -2.8181e+00],\n",
       "          [ 1.0612e-01,  1.7205e-01,  2.4960e-01,  ...,  5.0836e-01,\n",
       "           -3.3402e-01,  4.0658e-01],\n",
       "          ...,\n",
       "          [-1.7519e-01, -1.0355e-01, -9.3071e-02,  ..., -8.7347e-02,\n",
       "            2.5317e-02,  4.5036e-01],\n",
       "          [-4.0592e-01, -1.0559e-01,  4.1126e-01,  ...,  2.5422e-01,\n",
       "           -1.9175e-01, -2.4327e-01],\n",
       "          [-5.3149e-01, -3.8790e-01,  1.0591e+00,  ..., -1.0451e-01,\n",
       "           -4.5546e-01, -4.9864e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 6.7713e-01,  1.3700e-01, -1.4624e-01,  ..., -1.2465e-01,\n",
       "           -1.1514e+00,  4.5951e-01],\n",
       "          [-4.7617e-02,  6.9281e-02, -1.6746e-01,  ..., -2.4540e-01,\n",
       "           -1.5614e-01,  7.1871e-01],\n",
       "          [ 2.5556e-01, -1.4108e-01, -3.5688e-01,  ...,  1.9315e-01,\n",
       "            6.2028e-01, -7.1118e-01],\n",
       "          ...,\n",
       "          [-1.0392e+00,  2.9743e-02, -2.7681e-01,  ..., -1.0712e-01,\n",
       "           -8.1458e-02,  3.3507e-01],\n",
       "          [ 1.2229e-03, -6.9733e-02, -8.6710e-01,  ..., -1.0833e-01,\n",
       "           -2.1319e-01, -3.0159e-01],\n",
       "          [-5.7169e-02,  4.4104e-01, -3.7601e-01,  ..., -1.5505e-01,\n",
       "           -6.0715e-01,  5.4562e-01]],\n",
       "\n",
       "         [[-2.7077e-01, -4.9132e-01, -2.6427e-01,  ...,  2.7851e-01,\n",
       "            5.4097e-01,  1.7035e-02],\n",
       "          [-2.6242e-01,  8.8092e-02,  1.2250e-01,  ...,  4.3706e-01,\n",
       "           -8.6507e-01, -1.0450e+00],\n",
       "          [-4.4647e-01, -4.3027e-01, -3.9230e-01,  ..., -4.8195e-01,\n",
       "           -6.0773e-02,  5.0567e-01],\n",
       "          ...,\n",
       "          [ 4.8199e-01, -1.4669e-01, -5.4196e-01,  ..., -4.8256e-01,\n",
       "           -4.9357e-01,  3.1740e-01],\n",
       "          [-3.4082e-01, -5.1056e-02, -6.7363e-01,  ...,  7.0704e-01,\n",
       "           -1.0340e+00, -2.3532e-01],\n",
       "          [-4.8953e-01, -4.2183e-01,  5.9949e-01,  ...,  8.6709e-01,\n",
       "            3.5712e-01, -1.2440e-01]],\n",
       "\n",
       "         [[-2.4038e-02,  8.0506e-01,  2.2715e-01,  ...,  1.6607e-01,\n",
       "           -4.3557e-01, -9.7678e-01],\n",
       "          [-1.2412e-01, -4.2715e-01, -2.7043e-02,  ...,  3.4613e-01,\n",
       "            8.4131e-01,  5.3655e-02],\n",
       "          [ 6.8721e-02,  1.2614e-01, -2.6011e-01,  ...,  7.9046e-01,\n",
       "           -7.5879e-02,  3.1325e-02],\n",
       "          ...,\n",
       "          [-4.9727e-01, -3.8818e-05, -3.2518e-01,  ..., -4.0126e-01,\n",
       "           -2.7977e-01, -5.2358e-01],\n",
       "          [ 1.6245e-01,  1.4505e-01, -6.8052e-01,  ..., -9.3158e-01,\n",
       "           -1.0151e-01, -1.6348e-01],\n",
       "          [ 5.1066e-01,  6.0018e-01, -3.7637e-01,  ..., -2.9554e-01,\n",
       "            5.3153e-01, -1.2846e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.5252e-01, -9.0090e-01, -5.2641e-01,  ..., -4.5187e-01,\n",
       "            3.9341e-01, -5.6938e-01],\n",
       "          [-7.9991e-02, -3.0969e-01,  1.9610e-01,  ..., -2.1270e-01,\n",
       "           -5.6228e-02, -1.3635e-01],\n",
       "          [ 2.6994e-01,  3.2342e-01,  7.0271e-01,  ..., -2.9271e-01,\n",
       "            5.0884e-01,  9.7580e-01],\n",
       "          ...,\n",
       "          [ 3.5721e-01,  8.2532e-04, -2.2391e-01,  ...,  2.6261e-02,\n",
       "            4.3314e-01,  3.2739e-01],\n",
       "          [-5.0900e-01, -1.1300e-01, -1.0323e-01,  ..., -1.2354e-02,\n",
       "            2.4488e-01,  3.5775e-01],\n",
       "          [-5.7675e-02,  5.1790e-01, -9.6950e-02,  ..., -2.3500e-01,\n",
       "           -1.3399e-01,  4.6183e-01]],\n",
       "\n",
       "         [[-2.8412e-01, -2.7054e-02,  1.6716e-02,  ...,  5.4450e-02,\n",
       "            1.4047e-02,  1.3345e-01],\n",
       "          [ 5.0121e-01, -1.1093e-02, -1.2862e-01,  ...,  8.6384e-02,\n",
       "            3.1097e-02, -1.4725e-01],\n",
       "          [-4.4999e-01, -6.0789e-01, -1.0159e-01,  ..., -2.0347e-01,\n",
       "            2.5184e-02,  2.5388e-01],\n",
       "          ...,\n",
       "          [-4.5277e-02, -8.4614e-01, -7.5960e-01,  ..., -2.1637e-01,\n",
       "            7.0492e-01,  2.4995e-01],\n",
       "          [ 1.3085e-01, -8.8821e-01, -3.2170e-01,  ..., -1.6020e-01,\n",
       "           -3.1835e-01,  2.9364e-01],\n",
       "          [ 1.1330e-01, -6.6315e-01,  1.1003e-01,  ..., -3.9979e-01,\n",
       "           -1.4160e-01,  2.2665e-01]],\n",
       "\n",
       "         [[-1.0759e-01,  4.2959e-01,  5.5070e-01,  ..., -7.0609e-01,\n",
       "            1.0711e-01, -7.0728e-01],\n",
       "          [ 4.1106e-01,  8.0494e-01,  4.9595e-01,  ..., -1.3576e+00,\n",
       "            6.3415e-01, -6.5624e-01],\n",
       "          [-1.4029e-01, -1.8952e-01, -1.3221e-01,  ..., -1.2492e-01,\n",
       "           -2.2788e-01,  6.8498e-02],\n",
       "          ...,\n",
       "          [-1.4274e-01, -7.4864e-01,  7.6342e-01,  ..., -4.5911e-01,\n",
       "           -9.7082e-02, -2.9114e-01],\n",
       "          [-2.6067e-01,  2.9913e-01,  3.2762e-01,  ...,  8.4270e-01,\n",
       "            1.0533e-01,  7.2592e-01],\n",
       "          [-6.9971e-03, -2.0816e-01,  1.6942e-01,  ..., -5.8976e-01,\n",
       "           -6.0270e-01, -9.0385e-02]]],\n",
       "\n",
       "\n",
       "        [[[ 6.7713e-01,  1.3700e-01, -1.4624e-01,  ..., -1.2465e-01,\n",
       "           -1.1514e+00,  4.5951e-01],\n",
       "          [-4.7617e-02,  6.9281e-02, -1.6746e-01,  ..., -2.4540e-01,\n",
       "           -1.5614e-01,  7.1871e-01],\n",
       "          [ 1.2575e-01, -2.4779e-01, -3.3024e-01,  ..., -2.1509e-01,\n",
       "            2.8337e-01, -2.2901e-01],\n",
       "          ...,\n",
       "          [ 3.5008e-01,  2.8968e-01, -1.4720e-01,  ..., -3.0627e-02,\n",
       "           -1.4253e-01,  7.5975e-01],\n",
       "          [ 1.2229e-03, -6.9733e-02, -8.6710e-01,  ..., -1.0833e-01,\n",
       "           -2.1319e-01, -3.0159e-01],\n",
       "          [-1.0886e-01, -6.6319e-01, -3.8259e-01,  ...,  1.9730e-01,\n",
       "           -1.6876e-01, -2.9380e-01]],\n",
       "\n",
       "         [[-2.7077e-01, -4.9132e-01, -2.6427e-01,  ...,  2.7851e-01,\n",
       "            5.4097e-01,  1.7035e-02],\n",
       "          [-2.6242e-01,  8.8092e-02,  1.2250e-01,  ...,  4.3706e-01,\n",
       "           -8.6507e-01, -1.0450e+00],\n",
       "          [ 7.5493e-01, -1.1697e-02,  3.1113e-01,  ..., -7.4981e-02,\n",
       "           -1.1413e-01,  6.7462e-01],\n",
       "          ...,\n",
       "          [ 5.6538e-01,  3.7859e-01,  2.9059e-02,  ...,  5.0298e-01,\n",
       "            7.8298e-01, -2.7955e-03],\n",
       "          [-3.4082e-01, -5.1056e-02, -6.7363e-01,  ...,  7.0704e-01,\n",
       "           -1.0340e+00, -2.3532e-01],\n",
       "          [ 4.9496e-01, -9.8838e-01, -5.8103e-01,  ..., -5.0607e-02,\n",
       "           -7.2347e-01, -5.3571e-01]],\n",
       "\n",
       "         [[-2.4038e-02,  8.0506e-01,  2.2715e-01,  ...,  1.6607e-01,\n",
       "           -4.3557e-01, -9.7678e-01],\n",
       "          [-1.2412e-01, -4.2715e-01, -2.7043e-02,  ...,  3.4613e-01,\n",
       "            8.4131e-01,  5.3655e-02],\n",
       "          [ 4.4765e-01,  7.2418e-01,  6.0820e-01,  ..., -7.6388e-01,\n",
       "            2.4413e-01,  5.8513e-01],\n",
       "          ...,\n",
       "          [ 1.4863e-01,  3.0252e-01,  2.6000e-01,  ...,  3.9296e-01,\n",
       "           -2.8198e-01,  2.0196e-02],\n",
       "          [ 1.6245e-01,  1.4505e-01, -6.8052e-01,  ..., -9.3158e-01,\n",
       "           -1.0151e-01, -1.6348e-01],\n",
       "          [ 1.4455e-01,  7.9514e-01,  2.1737e-01,  ..., -3.5059e-02,\n",
       "           -5.5899e-01, -1.0949e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.5252e-01, -9.0090e-01, -5.2641e-01,  ..., -4.5187e-01,\n",
       "            3.9341e-01, -5.6938e-01],\n",
       "          [-7.9991e-02, -3.0969e-01,  1.9610e-01,  ..., -2.1270e-01,\n",
       "           -5.6228e-02, -1.3635e-01],\n",
       "          [ 1.1395e+00,  1.7468e+00, -3.9984e-01,  ..., -3.9215e-01,\n",
       "           -1.1977e-01, -7.4688e-01],\n",
       "          ...,\n",
       "          [ 4.7191e-01, -5.5488e-01,  6.3612e-02,  ...,  4.0132e-01,\n",
       "           -1.3858e-01,  5.1892e-01],\n",
       "          [-5.0900e-01, -1.1300e-01, -1.0323e-01,  ..., -1.2354e-02,\n",
       "            2.4488e-01,  3.5775e-01],\n",
       "          [-3.0930e-01,  1.4357e-01,  2.0217e-01,  ...,  3.6451e-01,\n",
       "           -8.1117e-01, -6.7174e-01]],\n",
       "\n",
       "         [[-2.8412e-01, -2.7054e-02,  1.6716e-02,  ...,  5.4450e-02,\n",
       "            1.4047e-02,  1.3345e-01],\n",
       "          [ 5.0121e-01, -1.1093e-02, -1.2862e-01,  ...,  8.6384e-02,\n",
       "            3.1097e-02, -1.4725e-01],\n",
       "          [-2.3596e-01,  2.1636e-02, -1.9700e-01,  ...,  3.9392e-01,\n",
       "            7.5116e-01, -4.8518e-01],\n",
       "          ...,\n",
       "          [ 1.4057e-01, -5.8168e-01, -8.0405e-01,  ..., -5.5158e-01,\n",
       "            5.8051e-02, -2.3034e-02],\n",
       "          [ 1.3085e-01, -8.8821e-01, -3.2170e-01,  ..., -1.6020e-01,\n",
       "           -3.1835e-01,  2.9364e-01],\n",
       "          [ 7.7390e-01,  4.7680e-01, -2.4749e-01,  ..., -1.7650e-01,\n",
       "            1.5175e-01, -4.1581e-01]],\n",
       "\n",
       "         [[-1.0759e-01,  4.2959e-01,  5.5070e-01,  ..., -7.0609e-01,\n",
       "            1.0711e-01, -7.0728e-01],\n",
       "          [ 4.1106e-01,  8.0494e-01,  4.9595e-01,  ..., -1.3576e+00,\n",
       "            6.3415e-01, -6.5624e-01],\n",
       "          [ 7.0116e-02,  5.2914e-01,  2.4537e-01,  ..., -3.2939e-02,\n",
       "           -3.1883e-01,  3.3416e-02],\n",
       "          ...,\n",
       "          [-5.1768e-01,  6.2548e-01,  2.0550e-01,  ..., -1.7932e-01,\n",
       "           -9.9592e-02, -7.7843e-02],\n",
       "          [-2.6067e-01,  2.9913e-01,  3.2762e-01,  ...,  8.4270e-01,\n",
       "            1.0533e-01,  7.2592e-01],\n",
       "          [ 1.4105e-01, -5.7161e-01, -8.6958e-01,  ...,  4.3032e-01,\n",
       "            7.4000e-01,  4.6118e-02]]],\n",
       "\n",
       "\n",
       "        [[[ 6.7713e-01,  1.3700e-01, -1.4624e-01,  ..., -1.2465e-01,\n",
       "           -1.1514e+00,  4.5951e-01],\n",
       "          [-4.7617e-02,  6.9281e-02, -1.6746e-01,  ..., -2.4540e-01,\n",
       "           -1.5614e-01,  7.1871e-01],\n",
       "          [ 2.4890e-01, -3.0171e-01, -1.0752e+00,  ...,  2.7185e-01,\n",
       "           -4.8481e-01,  1.6908e+00],\n",
       "          ...,\n",
       "          [ 1.7329e-01,  4.1646e-01,  1.9265e-01,  ..., -3.3174e-01,\n",
       "            8.1741e-01,  3.7945e-01],\n",
       "          [-5.4777e-01, -6.1943e-01, -1.9811e+00,  ...,  7.5718e-01,\n",
       "           -2.2608e+00, -1.8892e+00],\n",
       "          [-1.3348e-01,  3.0202e-01,  4.0452e-01,  ...,  4.5919e-01,\n",
       "            2.4672e-01,  4.3936e-01]],\n",
       "\n",
       "         [[-2.7077e-01, -4.9132e-01, -2.6427e-01,  ...,  2.7851e-01,\n",
       "            5.4097e-01,  1.7035e-02],\n",
       "          [-2.6242e-01,  8.8092e-02,  1.2250e-01,  ...,  4.3706e-01,\n",
       "           -8.6507e-01, -1.0450e+00],\n",
       "          [-6.0929e-01, -8.0120e-02,  1.8039e-01,  ...,  3.5563e-01,\n",
       "           -4.5929e-01, -1.5815e+00],\n",
       "          ...,\n",
       "          [-5.5708e-01,  5.8500e-01,  3.4737e-02,  ..., -4.6487e-01,\n",
       "            5.1573e-02, -1.5101e-01],\n",
       "          [ 9.3271e-01, -9.8333e-01, -1.1641e+00,  ..., -2.8199e-01,\n",
       "           -4.3309e-01, -3.3835e-01],\n",
       "          [-5.6786e-01,  4.3040e-01, -5.1533e-02,  ...,  6.1269e-01,\n",
       "            7.2876e-01,  4.3575e-01]],\n",
       "\n",
       "         [[-2.4038e-02,  8.0506e-01,  2.2715e-01,  ...,  1.6607e-01,\n",
       "           -4.3557e-01, -9.7678e-01],\n",
       "          [-1.2412e-01, -4.2715e-01, -2.7043e-02,  ...,  3.4613e-01,\n",
       "            8.4131e-01,  5.3655e-02],\n",
       "          [ 5.4194e-01,  7.5056e-01, -3.0018e-01,  ..., -1.2520e+00,\n",
       "            2.7594e-01, -6.8328e-01],\n",
       "          ...,\n",
       "          [-3.9695e-01, -3.5280e-03,  4.1151e-01,  ...,  1.2208e-01,\n",
       "           -1.2470e+00,  5.3350e-02],\n",
       "          [ 1.5659e-01,  8.1530e-01,  1.5722e-01,  ..., -1.3640e+00,\n",
       "           -6.0900e-02, -9.9367e-01],\n",
       "          [-3.3375e-01, -1.8408e-01,  3.1304e-01,  ...,  4.5653e-01,\n",
       "           -2.5922e-01,  2.8056e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.5252e-01, -9.0090e-01, -5.2641e-01,  ..., -4.5187e-01,\n",
       "            3.9341e-01, -5.6938e-01],\n",
       "          [-7.9991e-02, -3.0969e-01,  1.9610e-01,  ..., -2.1270e-01,\n",
       "           -5.6228e-02, -1.3635e-01],\n",
       "          [-7.3701e-01,  8.9703e-02, -3.5260e-01,  ...,  1.9421e-01,\n",
       "            3.6209e-01, -3.4558e-01],\n",
       "          ...,\n",
       "          [ 4.4121e-01,  6.5371e-01, -9.5309e-02,  ...,  1.0182e-01,\n",
       "           -5.0677e-01,  9.1857e-02],\n",
       "          [ 5.1618e-02, -5.1331e-02,  1.2382e-01,  ...,  7.6087e-02,\n",
       "           -4.3025e-01, -1.4528e-01],\n",
       "          [ 2.0371e-01,  6.4084e-01, -4.5484e-01,  ...,  4.2767e-02,\n",
       "            1.6893e-01, -5.4823e-01]],\n",
       "\n",
       "         [[-2.8412e-01, -2.7054e-02,  1.6716e-02,  ...,  5.4450e-02,\n",
       "            1.4047e-02,  1.3345e-01],\n",
       "          [ 5.0121e-01, -1.1093e-02, -1.2862e-01,  ...,  8.6384e-02,\n",
       "            3.1097e-02, -1.4725e-01],\n",
       "          [ 1.5281e+00, -5.2429e-01,  4.1076e-01,  ...,  3.2494e-01,\n",
       "           -1.3795e+00,  7.0282e-02],\n",
       "          ...,\n",
       "          [ 2.1392e-01, -1.6844e-01,  4.0749e-01,  ..., -2.9347e-01,\n",
       "           -4.5689e-01, -3.5649e-01],\n",
       "          [ 2.7065e-01, -7.7638e-01, -1.6290e-01,  ...,  1.5088e-01,\n",
       "           -8.9039e-02,  6.4153e-01],\n",
       "          [-1.0893e-01,  2.0202e-01,  1.1338e-01,  ...,  2.0921e-01,\n",
       "            6.3453e-01,  3.6557e-01]],\n",
       "\n",
       "         [[-1.0759e-01,  4.2959e-01,  5.5070e-01,  ..., -7.0609e-01,\n",
       "            1.0711e-01, -7.0728e-01],\n",
       "          [ 4.1106e-01,  8.0494e-01,  4.9595e-01,  ..., -1.3576e+00,\n",
       "            6.3415e-01, -6.5624e-01],\n",
       "          [-3.8487e-01,  8.0353e-02, -3.9886e-01,  ..., -1.0776e-01,\n",
       "            3.9551e-01, -3.0211e-02],\n",
       "          ...,\n",
       "          [-3.5932e-02, -2.0198e-01,  2.6032e-01,  ..., -4.1192e-02,\n",
       "           -2.5902e-01,  6.0093e-01],\n",
       "          [ 2.3304e-01,  3.2945e-01,  7.6915e-01,  ..., -8.0178e-02,\n",
       "            1.7313e-01,  1.1550e-01],\n",
       "          [-3.4285e-01, -2.3008e-01,  5.0316e-01,  ..., -9.1242e-01,\n",
       "            5.3776e-01,  4.1137e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.6676e+00, -2.1796e+00, -5.4693e-01,  ...,  2.7562e-01,\n",
       "           -1.4438e+00, -1.8213e+00],\n",
       "          [-1.2129e+00, -3.0597e-01,  1.1671e+00,  ..., -9.6079e-01,\n",
       "           -2.4931e+00,  5.0594e-01],\n",
       "          [-1.3639e+00,  1.4589e-01, -1.0199e+00,  ..., -2.4538e-01,\n",
       "           -1.2605e+00,  2.5509e-01],\n",
       "          ...,\n",
       "          [ 7.7639e-01, -1.0285e+00,  8.0949e-01,  ...,  3.8732e-01,\n",
       "           -1.3861e+00,  1.5115e-01],\n",
       "          [ 7.5447e-01, -1.3773e+00,  1.6870e+00,  ...,  8.1429e-01,\n",
       "           -1.0193e+00, -9.9350e-01],\n",
       "          [-6.1491e-01, -1.7505e-01,  1.6895e+00,  ...,  1.6422e+00,\n",
       "           -1.1416e+00, -2.1249e-01]],\n",
       "\n",
       "         [[ 1.0288e+00, -7.0680e-01, -1.6393e+00,  ..., -3.9511e-01,\n",
       "           -5.5960e-01, -2.1135e+00],\n",
       "          [ 1.2025e+00, -1.4204e+00,  2.8119e-01,  ..., -1.0094e+00,\n",
       "           -9.3031e-01, -1.7200e+00],\n",
       "          [ 1.1337e+00,  8.2045e-01,  1.6324e-01,  ...,  3.4833e-01,\n",
       "           -2.1311e+00, -9.2719e-01],\n",
       "          ...,\n",
       "          [ 7.7669e-01, -1.7775e+00, -1.3729e+00,  ...,  1.5947e-01,\n",
       "           -2.9451e+00, -2.4947e+00],\n",
       "          [ 1.8854e-01, -4.1905e-01,  1.4714e-01,  ...,  3.2198e+00,\n",
       "            1.9118e-01, -1.1079e-01],\n",
       "          [-1.1114e+00, -2.8075e-01,  1.3075e+00,  ...,  2.1901e+00,\n",
       "            1.8470e+00, -2.6618e-01]],\n",
       "\n",
       "         [[ 8.8236e-02,  1.8829e+00, -1.3242e-01,  ..., -3.1244e-01,\n",
       "           -1.1493e+00,  1.1473e+00],\n",
       "          [-1.2079e-01,  2.7859e-01,  6.4778e-01,  ...,  2.4780e+00,\n",
       "            1.4790e+00,  1.2618e+00],\n",
       "          [-2.1244e+00, -3.0057e-01,  1.1533e+00,  ..., -2.3412e-01,\n",
       "            5.7012e-01,  2.1097e+00],\n",
       "          ...,\n",
       "          [-7.8690e-01, -2.4306e-01, -7.3093e-01,  ...,  3.9536e-01,\n",
       "           -6.0948e-01, -6.7219e-02],\n",
       "          [ 2.7736e-01, -1.0403e+00,  1.3560e+00,  ..., -1.9201e-01,\n",
       "           -1.2917e-02,  1.0067e+00],\n",
       "          [ 2.2294e+00, -2.4819e+00,  2.4205e+00,  ...,  5.3300e-02,\n",
       "            8.7348e-01, -1.3506e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 6.6650e-01,  3.1997e-01, -1.1894e-01,  ..., -1.5015e+00,\n",
       "           -1.4693e+00, -1.3812e+00],\n",
       "          [-1.9938e+00, -7.0797e-02,  1.8453e-02,  ...,  2.6391e+00,\n",
       "            9.7365e-01,  1.2230e+00],\n",
       "          [-2.2238e+00, -7.0883e-01,  9.7680e-02,  ...,  1.5120e+00,\n",
       "            2.9604e-01,  2.3396e-01],\n",
       "          ...,\n",
       "          [-2.8292e+00, -7.2962e-01, -7.3996e-01,  ...,  3.3829e-02,\n",
       "            7.3708e-01,  1.2699e+00],\n",
       "          [-6.2854e-01, -1.0082e+00,  1.2891e+00,  ...,  7.7403e-01,\n",
       "           -1.9881e-01, -1.9926e+00],\n",
       "          [ 8.7464e-01, -1.9482e+00, -1.5071e+00,  ...,  1.4144e-01,\n",
       "            1.0128e+00, -1.2084e+00]],\n",
       "\n",
       "         [[ 3.2788e-01,  1.6972e+00,  3.2699e+00,  ...,  2.0325e-01,\n",
       "           -8.0808e-01,  9.1688e-01],\n",
       "          [-2.0023e-01,  1.0500e+00, -1.2801e+00,  ..., -2.8048e-01,\n",
       "            1.9310e-01,  1.2449e+00],\n",
       "          [-1.6715e+00,  4.1514e-01,  9.6586e-02,  ...,  8.6864e-01,\n",
       "            1.1864e+00,  1.1590e+00],\n",
       "          ...,\n",
       "          [ 1.7942e-01, -1.0631e+00,  6.4583e-01,  ..., -1.4514e+00,\n",
       "            2.2087e+00, -7.2577e-01],\n",
       "          [-8.6297e-01,  4.5969e-01,  4.1522e-01,  ...,  1.1997e+00,\n",
       "           -1.4463e+00,  3.3499e-01],\n",
       "          [-8.3318e-01, -2.9986e-01,  1.0458e+00,  ..., -1.4793e+00,\n",
       "           -5.2413e-01,  1.5470e+00]],\n",
       "\n",
       "         [[ 1.2533e+00, -6.5112e-01, -1.2189e+00,  ...,  4.1791e-01,\n",
       "           -4.1700e-02, -1.2730e+00],\n",
       "          [ 1.8177e+00,  5.2951e-02,  2.2750e-01,  ..., -1.4276e+00,\n",
       "            7.0812e-01,  2.6922e-01],\n",
       "          [ 8.4254e-01,  1.4761e-01,  1.5434e-01,  ..., -6.5349e-01,\n",
       "            4.7618e-01, -7.6721e-01],\n",
       "          ...,\n",
       "          [ 2.5049e+00,  7.1775e-01,  8.9926e-02,  ...,  8.7435e-03,\n",
       "            6.8375e-01, -1.6900e+00],\n",
       "          [ 1.5071e-01,  1.0318e+00, -1.6850e+00,  ...,  1.6648e+00,\n",
       "            8.1393e-01,  8.7139e-01],\n",
       "          [-1.7866e+00, -1.1575e+00,  4.6549e-02,  ...,  1.1279e+00,\n",
       "            1.5921e+00,  1.3048e+00]]],\n",
       "\n",
       "\n",
       "        [[[-2.0347e+00,  1.8267e-01,  1.3880e+00,  ...,  2.7686e+00,\n",
       "           -3.5465e+00, -1.1725e+00],\n",
       "          [-1.7171e+00,  6.9786e-01, -5.9997e-01,  ..., -2.2434e-01,\n",
       "           -1.9879e+00,  3.0881e-01],\n",
       "          [-2.4206e+00,  8.7588e-01,  9.1592e-02,  ...,  5.6587e-01,\n",
       "           -3.7011e+00, -7.5963e-01],\n",
       "          ...,\n",
       "          [ 1.3402e+00, -4.9258e-01,  2.1637e+00,  ...,  8.7167e-01,\n",
       "           -3.2002e+00,  1.5075e+00],\n",
       "          [-4.3405e-01, -8.4095e-01,  2.9623e+00,  ...,  2.2422e-02,\n",
       "           -4.9330e-01, -1.8076e-01],\n",
       "          [-6.3632e-01, -2.0321e-01,  1.6611e+00,  ...,  1.6083e+00,\n",
       "           -1.1189e+00, -1.3837e-01]],\n",
       "\n",
       "         [[ 8.1824e-02,  1.4034e+00, -7.3795e-01,  ..., -1.4979e+00,\n",
       "           -4.7715e-02, -1.0494e+00],\n",
       "          [-3.5961e-01,  1.3583e+00,  3.4002e-01,  ..., -4.5391e-01,\n",
       "           -1.6245e+00, -1.1178e+00],\n",
       "          [ 1.2179e-01,  2.2121e+00, -4.6040e-01,  ..., -5.7843e-01,\n",
       "           -4.0678e-01,  9.8229e-01],\n",
       "          ...,\n",
       "          [-3.3251e-01, -4.1355e-01, -5.5737e-02,  ...,  1.2878e+00,\n",
       "           -1.2185e+00, -1.6260e+00],\n",
       "          [-4.5591e-01,  3.3872e-01,  3.1423e-01,  ...,  3.0024e+00,\n",
       "            9.4466e-01, -2.7922e-01],\n",
       "          [-1.1435e+00, -3.1764e-01,  1.3028e+00,  ...,  2.2167e+00,\n",
       "            1.8540e+00, -2.6877e-01]],\n",
       "\n",
       "         [[-9.2288e-01,  9.5196e-01,  3.4640e-01,  ...,  1.9732e+00,\n",
       "            1.0904e+00,  6.9939e-01],\n",
       "          [-1.3806e+00, -1.2620e+00,  9.6556e-01,  ..., -9.6417e-02,\n",
       "            4.4137e-01,  1.2929e+00],\n",
       "          [-4.2043e-01, -4.2529e-01,  1.4193e+00,  ...,  4.8152e-02,\n",
       "            2.0894e-01,  6.4002e-01],\n",
       "          ...,\n",
       "          [-1.9621e+00,  1.0836e+00,  9.6888e-01,  ...,  1.6040e+00,\n",
       "           -5.0855e-01,  3.1397e-01],\n",
       "          [-1.5765e-01,  3.6802e-03,  1.3962e+00,  ..., -3.7690e-02,\n",
       "           -6.9015e-01,  6.6909e-01],\n",
       "          [ 2.1991e+00, -2.5397e+00,  2.4025e+00,  ...,  1.8979e-02,\n",
       "            8.2778e-01, -1.3519e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.9376e+00,  2.5196e-01, -1.5859e+00,  ..., -9.4173e-01,\n",
       "            2.7718e-01,  2.2447e+00],\n",
       "          [-2.2272e+00, -5.7786e-01,  3.6995e-01,  ...,  1.3921e+00,\n",
       "           -6.7804e-01,  1.0675e+00],\n",
       "          [-2.1691e+00,  1.9496e+00,  7.7681e-01,  ...,  1.4537e+00,\n",
       "            2.0896e-01, -9.7228e-01],\n",
       "          ...,\n",
       "          [-8.7363e-01, -2.3539e-02, -1.0323e+00,  ..., -8.8402e-01,\n",
       "            2.2058e-01,  3.4505e-01],\n",
       "          [-1.3597e+00, -2.4996e+00, -2.9887e-02,  ...,  1.0049e+00,\n",
       "            4.9959e-01, -2.0691e+00],\n",
       "          [ 8.5067e-01, -2.0156e+00, -1.5056e+00,  ...,  1.3184e-01,\n",
       "            1.0324e+00, -1.2397e+00]],\n",
       "\n",
       "         [[-9.0266e-01,  5.5824e-01,  1.3419e+00,  ..., -1.3645e+00,\n",
       "           -2.1565e-01,  1.5162e+00],\n",
       "          [-1.4411e+00,  7.2928e-01,  1.1277e+00,  ...,  1.1031e+00,\n",
       "            1.0364e+00, -6.9108e-02],\n",
       "          [-5.3096e-01,  1.6308e+00, -2.6949e-01,  ...,  1.2694e-01,\n",
       "            1.3682e+00,  1.0636e+00],\n",
       "          ...,\n",
       "          [-1.0952e-01, -1.6638e+00,  2.3340e+00,  ..., -6.1903e-01,\n",
       "           -2.1331e+00,  2.5113e+00],\n",
       "          [-1.3028e+00, -3.6842e-04,  1.0637e+00,  ...,  6.6260e-01,\n",
       "           -3.8826e-01,  1.5786e+00],\n",
       "          [-9.0352e-01, -3.3624e-01,  1.1109e+00,  ..., -1.4880e+00,\n",
       "           -5.3563e-01,  1.5223e+00]],\n",
       "\n",
       "         [[ 3.9943e+00,  2.0295e+00,  3.0785e+00,  ..., -1.2497e+00,\n",
       "            8.9431e-01, -1.0971e+00],\n",
       "          [ 7.9974e-01,  6.5222e-01,  1.0414e+00,  ..., -1.0858e+00,\n",
       "           -5.3223e-02, -7.4642e-01],\n",
       "          [ 1.2873e+00,  1.0761e+00,  9.1935e-01,  ..., -2.4909e+00,\n",
       "            5.6503e-01,  6.8142e-02],\n",
       "          ...,\n",
       "          [ 2.0938e+00,  2.1509e-01, -1.2882e+00,  ..., -1.8445e+00,\n",
       "           -1.0382e+00, -2.4969e+00],\n",
       "          [-5.0747e-01,  6.4099e-01, -1.7510e+00,  ...,  1.0985e+00,\n",
       "           -6.1415e-01,  4.2188e-01],\n",
       "          [-1.8318e+00, -1.1698e+00,  8.5398e-02,  ...,  1.1216e+00,\n",
       "            1.5721e+00,  1.2781e+00]]],\n",
       "\n",
       "\n",
       "        [[[-9.3921e-01, -2.0174e+00,  1.0156e+00,  ...,  3.5652e-01,\n",
       "           -1.3803e+00,  3.5786e-02],\n",
       "          [-3.9031e+00, -3.5365e-02,  1.1013e-01,  ...,  5.8012e-01,\n",
       "           -4.2382e-01, -2.0687e+00],\n",
       "          [-8.3009e-01,  5.2875e-01, -9.0250e-01,  ...,  1.7794e+00,\n",
       "           -2.2011e+00, -1.7076e+00],\n",
       "          ...,\n",
       "          [-1.2771e-01,  2.8535e-01,  8.4249e-01,  ...,  8.2597e-01,\n",
       "           -1.2877e+00,  8.2116e-01],\n",
       "          [ 3.2247e-01, -7.3277e-01,  2.5120e+00,  ...,  1.1886e+00,\n",
       "           -1.4039e+00, -5.7908e-01],\n",
       "          [-7.0203e-01, -1.8035e-01,  1.6585e+00,  ...,  1.6247e+00,\n",
       "           -1.1079e+00, -2.2914e-01]],\n",
       "\n",
       "         [[-4.8427e-01,  9.5545e-01, -2.4689e+00,  ..., -6.4423e-01,\n",
       "            7.9434e-01, -2.6787e-02],\n",
       "          [ 2.1234e+00,  1.0422e+00, -3.7195e-01,  ..., -1.3933e+00,\n",
       "           -8.6316e-01,  5.4155e-01],\n",
       "          [ 1.0701e+00,  9.0254e-01, -1.0370e+00,  ..., -6.4752e-01,\n",
       "            1.0190e+00, -1.4341e+00],\n",
       "          ...,\n",
       "          [-5.5747e-01,  3.4265e-01,  8.9567e-01,  ...,  5.2177e-01,\n",
       "            1.0539e+00,  1.4984e-02],\n",
       "          [-1.7530e+00, -7.6375e-01,  3.0578e-01,  ...,  2.3673e+00,\n",
       "            7.0853e-01, -4.1368e-01],\n",
       "          [-1.1195e+00, -2.6722e-01,  1.3190e+00,  ...,  2.1629e+00,\n",
       "            1.8811e+00, -2.5751e-01]],\n",
       "\n",
       "         [[ 8.2232e-01, -1.6387e+00,  1.0451e+00,  ..., -1.0101e+00,\n",
       "           -2.0937e+00,  2.3720e+00],\n",
       "          [-9.9868e-01,  1.3947e+00,  3.7661e-01,  ..., -7.6436e-02,\n",
       "           -8.9148e-01,  2.0816e+00],\n",
       "          [ 7.3790e-01, -1.3824e-01, -1.2156e+00,  ..., -2.9862e-01,\n",
       "           -1.8608e+00,  1.5735e+00],\n",
       "          ...,\n",
       "          [ 8.1714e-01, -6.4412e-01, -1.1867e+00,  ..., -2.8062e-01,\n",
       "           -1.0457e+00,  2.9778e-03],\n",
       "          [ 9.7045e-01, -4.5802e-01,  1.1567e+00,  ..., -2.6219e-01,\n",
       "           -5.0948e-01,  1.0568e+00],\n",
       "          [ 2.2598e+00, -2.5614e+00,  2.4317e+00,  ...,  4.3489e-02,\n",
       "            8.6296e-01, -1.3403e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.9713e+00,  1.7454e-01, -2.2422e+00,  ...,  3.2156e+00,\n",
       "           -2.9369e-01,  6.5621e-02],\n",
       "          [-3.6750e+00, -1.0299e+00, -3.2576e-01,  ...,  1.8954e+00,\n",
       "           -1.6074e+00, -5.8887e-01],\n",
       "          [-9.8210e-01, -8.9486e-01,  6.6837e-01,  ...,  1.5950e+00,\n",
       "           -3.8369e-01,  8.8116e-01],\n",
       "          ...,\n",
       "          [-1.1518e+00, -2.8508e-01,  1.1412e+00,  ...,  1.5225e+00,\n",
       "            1.5011e+00, -1.5867e+00],\n",
       "          [-4.2341e-01, -1.3452e+00,  1.6700e+00,  ...,  8.8060e-01,\n",
       "           -1.6995e-01, -2.7912e+00],\n",
       "          [ 8.4559e-01, -2.0055e+00, -1.5302e+00,  ...,  2.0669e-01,\n",
       "            1.0080e+00, -1.2234e+00]],\n",
       "\n",
       "         [[ 7.3200e-01, -5.3482e-01,  1.9176e+00,  ...,  9.0605e-02,\n",
       "            1.7507e+00,  5.1217e-01],\n",
       "          [-1.8932e-01,  2.4000e-01,  1.4774e+00,  ...,  1.5993e+00,\n",
       "            1.2441e+00, -2.8340e+00],\n",
       "          [-7.8379e-01,  4.2979e-01,  1.2228e+00,  ..., -1.4667e-01,\n",
       "            2.6934e+00, -2.1256e+00],\n",
       "          ...,\n",
       "          [-6.6340e-01, -8.6516e-01, -4.2783e-01,  ...,  1.3641e+00,\n",
       "            1.5171e+00, -9.4095e-01],\n",
       "          [ 3.2864e-01, -1.5278e+00,  7.4396e-01,  ...,  1.0652e+00,\n",
       "           -4.1498e-01, -5.9414e-01],\n",
       "          [-8.7008e-01, -3.1074e-01,  1.0822e+00,  ..., -1.5328e+00,\n",
       "           -5.0491e-01,  1.5073e+00]],\n",
       "\n",
       "         [[ 1.9239e+00, -2.8855e-01,  1.2702e+00,  ..., -5.8344e-02,\n",
       "           -1.1519e+00, -2.7081e-01],\n",
       "          [ 6.8126e-01, -1.4466e+00,  1.2855e+00,  ...,  5.3176e-02,\n",
       "           -1.0943e+00,  2.1072e+00],\n",
       "          [ 3.2034e+00, -1.6277e+00,  6.0366e-01,  ..., -8.5146e-01,\n",
       "           -6.3894e-01, -5.4378e-01],\n",
       "          ...,\n",
       "          [ 2.1362e-01,  1.4721e+00, -8.9419e-01,  ..., -3.0907e+00,\n",
       "           -8.5083e-01,  1.0738e+00],\n",
       "          [ 1.0019e+00,  1.3275e+00, -3.0145e+00,  ...,  9.0665e-01,\n",
       "           -3.4926e-01,  5.3986e-01],\n",
       "          [-1.8154e+00, -1.1811e+00,  1.1881e-01,  ...,  1.1505e+00,\n",
       "            1.5560e+00,  1.3290e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 0.2228, -0.4759, -1.5214,  ...,  0.7449, -1.0441,  1.0512],\n",
       "          [-0.1673, -1.2263, -0.0847,  ...,  2.7759, -0.7536, -0.6058],\n",
       "          [ 0.0846, -0.9241, -0.0146,  ...,  1.0778, -1.9658, -1.0091],\n",
       "          ...,\n",
       "          [-0.0482, -1.3901, -1.0521,  ...,  2.5521,  0.1619, -0.6433],\n",
       "          [ 0.7274,  0.3371,  0.1104,  ..., -0.2507, -1.0697, -0.2395],\n",
       "          [-0.0249, -0.3279,  0.5156,  ...,  0.0668, -0.0648, -0.0572]],\n",
       "\n",
       "         [[-0.5285,  0.5164,  1.3259,  ..., -1.1225, -0.1388,  0.6870],\n",
       "          [ 0.6392,  0.1646,  1.1893,  ...,  0.8413,  0.0928,  0.7857],\n",
       "          [ 1.4782, -0.1840,  0.7437,  ...,  1.1944, -0.4546, -0.3613],\n",
       "          ...,\n",
       "          [-0.3966,  0.0784,  0.6027,  ..., -0.2228, -0.2851,  0.1021],\n",
       "          [-0.1675,  0.8497, -0.0149,  ..., -0.0344,  0.2768, -0.2122],\n",
       "          [-0.0435, -0.1627,  0.2599,  ...,  0.1358, -0.1885, -0.2455]],\n",
       "\n",
       "         [[ 0.3035,  0.6424,  0.6697,  ..., -1.5364,  0.6945,  0.8568],\n",
       "          [-0.4241,  0.5019,  0.1781,  ..., -0.0774, -0.7412, -0.3330],\n",
       "          [ 0.1440,  0.1717,  0.9016,  ..., -1.0362,  0.4915, -0.6716],\n",
       "          ...,\n",
       "          [-0.2911, -0.1061, -0.2307,  ..., -0.8350, -0.9809, -0.7256],\n",
       "          [ 0.5478, -0.5756,  0.5578,  ..., -0.5992, -0.2853, -0.3013],\n",
       "          [ 0.4507, -0.0957,  0.4208,  ...,  0.2354, -0.2043, -0.1591]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-0.6109,  1.1631, -1.2944,  ..., -0.4016,  1.2443,  2.6171],\n",
       "          [ 0.8617,  0.6169,  0.4644,  ...,  1.0990,  1.0083,  0.9515],\n",
       "          [ 0.0120,  0.9301, -0.8389,  ..., -0.1029,  0.4424, -0.6963],\n",
       "          ...,\n",
       "          [ 0.3791,  0.0260, -0.0815,  ..., -0.9887,  0.8765,  0.9515],\n",
       "          [-0.2338,  0.2582, -0.6469,  ..., -0.2904,  1.0313,  1.7049],\n",
       "          [-0.1422,  0.0157, -0.2770,  ..., -0.2096, -0.0766,  0.0639]],\n",
       "\n",
       "         [[-0.3255,  0.7185, -0.2346,  ...,  0.1396, -0.4285, -0.4392],\n",
       "          [ 0.3262, -0.9370,  0.7509,  ..., -0.6200, -0.1856, -0.9828],\n",
       "          [-1.0190,  0.5287, -0.4903,  ...,  0.8695, -0.9066, -0.4421],\n",
       "          ...,\n",
       "          [ 0.2865,  0.2406,  0.4188,  ...,  0.9244, -1.3396, -1.2602],\n",
       "          [ 0.2272,  0.1512, -0.2501,  ...,  0.9882, -0.1238, -0.5898],\n",
       "          [-0.1219, -0.0882,  0.0128,  ..., -0.0315, -0.1819, -0.1255]],\n",
       "\n",
       "         [[ 1.1271, -1.4614,  1.0098,  ..., -0.9968, -0.3725,  2.0785],\n",
       "          [-0.3532,  0.4493, -0.6742,  ..., -1.0158,  0.6183,  0.6671],\n",
       "          [-0.1913,  0.5855, -0.1288,  ..., -0.8827,  0.6254,  0.8577],\n",
       "          ...,\n",
       "          [-0.4102,  0.2507,  0.7370,  ..., -0.3319, -0.1549,  1.8980],\n",
       "          [ 0.0978, -0.0156,  0.8983,  ...,  0.3930, -0.7468,  0.6462],\n",
       "          [-0.1732,  0.1933,  0.4793,  ..., -0.2324, -0.0568,  0.2556]]],\n",
       "\n",
       "\n",
       "        [[[ 1.2075,  0.0141,  0.8200,  ...,  2.4052, -0.0165,  0.5244],\n",
       "          [ 0.7447,  0.4258, -0.1654,  ...,  1.4119, -1.2567, -0.7302],\n",
       "          [ 2.5097,  1.1530,  0.7259,  ...,  0.3326,  0.3048,  0.1175],\n",
       "          ...,\n",
       "          [-0.0189, -0.3305, -1.3907,  ...,  0.2774,  0.2328, -1.8722],\n",
       "          [ 0.8899,  0.5490, -0.3074,  ..., -0.1570, -0.4147, -0.4897],\n",
       "          [-0.0415, -0.3209,  0.5385,  ...,  0.0741, -0.0576, -0.0688]],\n",
       "\n",
       "         [[-0.2658, -0.0809, -0.3161,  ..., -0.1644, -1.7447,  0.6419],\n",
       "          [ 1.1340, -0.9608, -0.1978,  ...,  1.1644, -0.5108, -0.4067],\n",
       "          [ 0.7082, -0.5342, -0.4737,  ...,  1.4685,  1.2636, -0.5224],\n",
       "          ...,\n",
       "          [-0.0978, -1.2015,  0.3844,  ...,  0.1118, -0.4913,  0.1343],\n",
       "          [ 0.0065,  0.8622,  0.6745,  ..., -1.2594, -0.0633, -0.5170],\n",
       "          [-0.0264, -0.1571,  0.2769,  ...,  0.1546, -0.2043, -0.2259]],\n",
       "\n",
       "         [[-0.2640, -0.5343,  1.1886,  ...,  0.7680,  0.3614, -0.5877],\n",
       "          [-0.1159, -0.3491,  0.1759,  ..., -1.3690,  0.1430, -1.0251],\n",
       "          [-0.1224,  0.1615, -1.4942,  ..., -0.8880, -0.0445,  0.5022],\n",
       "          ...,\n",
       "          [ 0.6353, -0.6815,  0.5149,  ..., -1.0584,  1.9256, -0.9824],\n",
       "          [ 0.4038, -0.3867,  0.6363,  ..., -0.6367,  0.4031, -0.4339],\n",
       "          [ 0.4499, -0.1323,  0.4473,  ...,  0.2120, -0.2015, -0.1988]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-0.1974,  1.0720, -1.5995,  ..., -0.3416,  1.2825,  0.6969],\n",
       "          [-0.3150,  0.8967, -0.2856,  ..., -0.6839,  0.2910,  0.1444],\n",
       "          [ 0.2873,  0.6539,  0.7203,  ...,  0.4835,  0.8573,  1.0041],\n",
       "          ...,\n",
       "          [ 0.5757,  0.2796, -0.2808,  ...,  0.8123,  0.8505, -0.3635],\n",
       "          [ 0.6152,  0.4624, -0.7519,  ...,  0.4369,  0.3356,  0.3527],\n",
       "          [-0.1704,  0.0355, -0.2484,  ..., -0.1924, -0.0818,  0.0577]],\n",
       "\n",
       "         [[ 0.5882,  2.0352, -0.1034,  ...,  0.9473,  1.0425, -0.0470],\n",
       "          [-1.1689,  0.5475, -0.0052,  ...,  1.5653, -0.8154, -0.3891],\n",
       "          [ 0.0088,  0.7978, -1.3067,  ..., -0.4625, -1.0106, -0.3612],\n",
       "          ...,\n",
       "          [ 0.1581, -0.2572,  2.1491,  ..., -0.1102, -0.5477, -0.2028],\n",
       "          [ 0.0516, -0.4844,  0.2821,  ...,  0.8082, -0.0255, -0.6478],\n",
       "          [-0.1702, -0.0769,  0.0460,  ..., -0.0159, -0.1829, -0.1170]],\n",
       "\n",
       "         [[ 0.9756,  1.0383,  1.1839,  ..., -0.1362,  1.2212,  0.4092],\n",
       "          [ 0.1640,  0.6663,  0.5177,  ..., -0.2644,  0.4821,  0.5036],\n",
       "          [ 1.7356,  0.9249,  0.7877,  ..., -0.0310, -0.7055, -0.9471],\n",
       "          ...,\n",
       "          [ 0.4686,  0.2697, -0.5526,  ...,  0.0306,  0.0390,  0.8339],\n",
       "          [ 0.5067,  0.1256,  0.3844,  ..., -0.6740,  0.2473,  0.7614],\n",
       "          [-0.1557,  0.1903,  0.4526,  ..., -0.2803, -0.0398,  0.1983]]],\n",
       "\n",
       "\n",
       "        [[[ 1.1151,  1.0079, -0.8957,  ...,  0.3829, -0.8892, -0.5121],\n",
       "          [ 1.2813,  1.7762,  0.2826,  ...,  0.3212, -2.0112,  0.7151],\n",
       "          [-0.9000,  1.1987,  0.3085,  ..., -0.1238, -0.7172,  0.0371],\n",
       "          ...,\n",
       "          [ 1.0784,  1.2249, -0.6759,  ...,  0.0924,  0.5179, -1.2364],\n",
       "          [ 0.7786,  0.0150,  0.2640,  ..., -0.3530, -0.1694, -0.3173],\n",
       "          [-0.0083, -0.3203,  0.5568,  ...,  0.0610, -0.0502, -0.0429]],\n",
       "\n",
       "         [[-0.1933, -0.1869,  0.7612,  ..., -0.9763,  0.1499, -1.2746],\n",
       "          [-0.7123, -0.3907,  0.7579,  ...,  0.1998, -0.0831, -1.6803],\n",
       "          [-0.7738, -0.0566, -1.4370,  ...,  0.4234, -0.1979, -1.9665],\n",
       "          ...,\n",
       "          [-0.2776,  0.7307, -0.4692,  ..., -1.2287,  0.6474, -0.8526],\n",
       "          [-1.1736,  1.1172, -0.1300,  ...,  0.5586, -0.0185, -0.2316],\n",
       "          [-0.0547, -0.1430,  0.2659,  ...,  0.1341, -0.2346, -0.2303]],\n",
       "\n",
       "         [[-0.1846, -0.7261, -0.2597,  ...,  0.7117,  0.0622,  0.9912],\n",
       "          [-0.3527, -0.6765,  0.0555,  ...,  0.0238, -1.0561,  0.5913],\n",
       "          [-0.0480, -1.8622, -0.1971,  ...,  0.8737,  0.1123, -0.6654],\n",
       "          ...,\n",
       "          [-0.1961, -0.8836, -1.2468,  ...,  0.8592, -0.2457, -0.5815],\n",
       "          [ 0.2431, -0.6087, -0.3092,  ..., -0.0432, -0.0817, -0.4629],\n",
       "          [ 0.4393, -0.1293,  0.4174,  ...,  0.2358, -0.2029, -0.1627]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-0.7148, -0.0552, -0.2614,  ..., -0.1616, -0.1296, -0.2755],\n",
       "          [-0.8834,  0.4702, -0.1172,  ..., -1.3984,  0.6500, -1.0991],\n",
       "          [ 0.1749,  1.3178, -0.7799,  ..., -0.7401, -0.3120, -0.2154],\n",
       "          ...,\n",
       "          [ 0.4872, -1.0009, -0.4285,  ..., -0.3695,  0.4687, -0.5583],\n",
       "          [ 0.4541, -0.1007, -0.2619,  ..., -0.2225,  0.8099,  0.9178],\n",
       "          [-0.1669,  0.0368, -0.2678,  ..., -0.1791, -0.0745,  0.0520]],\n",
       "\n",
       "         [[-0.4440,  0.4583, -0.1299,  ..., -0.7631, -2.0791, -0.7705],\n",
       "          [ 0.7009,  1.2743, -0.6893,  ..., -1.1399, -0.7896,  0.2134],\n",
       "          [-0.5606, -0.0179,  2.1537,  ..., -0.4892, -0.1073, -0.7269],\n",
       "          ...,\n",
       "          [-0.5929,  0.3344, -0.3927,  ..., -0.6874, -0.3069, -1.0899],\n",
       "          [-0.2862,  0.4859, -0.6212,  ..., -0.6420, -0.3955, -0.3524],\n",
       "          [-0.1371, -0.0978,  0.0327,  ..., -0.0287, -0.1522, -0.1446]],\n",
       "\n",
       "         [[ 0.4605,  1.3751, -0.5559,  ..., -1.3347,  0.5497,  1.5076],\n",
       "          [ 0.1917, -0.6941, -0.8576,  ..., -0.2889,  1.0830, -0.6713],\n",
       "          [ 1.2626, -0.3838,  0.4443,  ..., -0.6148,  1.4147,  0.1820],\n",
       "          ...,\n",
       "          [ 0.9338, -0.2794,  0.8247,  ...,  0.1783, -0.8402,  1.3336],\n",
       "          [ 0.4858,  0.4504,  1.0373,  ...,  0.4224, -0.7979,  0.1329],\n",
       "          [-0.1538,  0.1678,  0.4310,  ..., -0.2239, -0.0314,  0.1901]]]],\n",
       "       grad_fn=<TransposeBackward0>)), (tensor([[[[-8.7621e-01, -5.0966e-01, -1.7272e+00,  ...,  2.4988e-01,\n",
       "           -8.7319e-01, -6.2302e-01],\n",
       "          [-2.2103e-01,  8.7154e-01,  4.0475e-01,  ..., -7.5032e-01,\n",
       "            9.8247e-01, -5.0854e-01],\n",
       "          [ 7.6457e-01, -9.4226e-01, -1.0263e+00,  ..., -3.8826e-01,\n",
       "            1.0832e+00, -8.6290e-01],\n",
       "          ...,\n",
       "          [ 2.0056e-01,  4.5206e-01, -1.0590e-01,  ...,  7.8712e-01,\n",
       "           -6.5034e-02, -1.6584e-01],\n",
       "          [-8.4760e-01,  2.7411e+00,  2.0242e+00,  ..., -2.3225e+00,\n",
       "            1.6314e+00,  2.4414e+00],\n",
       "          [-8.6201e-01, -6.5491e-01, -4.9598e-01,  ..., -1.6262e+00,\n",
       "            1.9347e-01,  8.6840e-01]],\n",
       "\n",
       "         [[-5.1410e-01,  4.9025e-02,  7.8485e-01,  ...,  6.9425e-01,\n",
       "           -1.0090e-01,  6.1841e-01],\n",
       "          [-6.1160e-01,  3.8945e-01, -2.5081e-01,  ...,  6.7742e-01,\n",
       "            6.4364e-01,  4.0325e-01],\n",
       "          [ 2.6101e-01,  4.5872e-01, -3.0882e-01,  ..., -1.4908e+00,\n",
       "           -1.0225e+00,  2.3136e-01],\n",
       "          ...,\n",
       "          [ 6.6327e-01,  2.7138e-01, -1.7048e+00,  ...,  3.2760e-01,\n",
       "           -4.9613e-02, -5.4612e-01],\n",
       "          [-1.8094e+00, -1.7837e+00, -8.7789e-01,  ...,  8.0100e-01,\n",
       "           -3.5783e-01, -1.0540e+00],\n",
       "          [-6.5074e-01,  6.4343e-01, -8.4612e-01,  ..., -6.6738e-01,\n",
       "            1.0614e+00, -8.6203e-01]],\n",
       "\n",
       "         [[ 1.1058e+00, -2.0079e+00, -4.4660e-01,  ..., -6.6941e-01,\n",
       "            1.8503e+00, -1.5254e+00],\n",
       "          [-2.5017e+00,  7.1920e-01,  2.1339e-01,  ...,  2.4825e+00,\n",
       "           -4.4840e-01,  2.2652e+00],\n",
       "          [-9.7736e-01,  6.5443e-01,  5.1702e-01,  ...,  6.8038e-01,\n",
       "           -6.7273e-01, -3.1306e-01],\n",
       "          ...,\n",
       "          [ 2.4659e-01, -6.3415e-01,  7.0418e-01,  ...,  1.7600e-01,\n",
       "           -8.7350e-01, -1.2646e+00],\n",
       "          [ 2.7300e+00, -3.2072e+00, -2.9487e+00,  ..., -2.2702e+00,\n",
       "            5.4085e-01, -2.8503e+00],\n",
       "          [ 4.4840e-01, -4.6073e-01, -5.2491e-02,  ..., -1.9248e-01,\n",
       "           -7.5747e-01, -1.0993e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.9580e-01,  5.9131e-01, -6.4563e-01,  ..., -4.4317e-01,\n",
       "           -1.4874e+00,  1.2797e+00],\n",
       "          [-4.4588e-01,  1.7430e+00, -1.9362e+00,  ...,  3.0873e-01,\n",
       "           -1.5742e+00,  7.3556e-01],\n",
       "          [ 2.6828e-01,  3.2418e-01,  1.2498e-01,  ..., -5.8194e-01,\n",
       "           -8.8225e-01,  7.7612e-01],\n",
       "          ...,\n",
       "          [ 8.5311e-01,  5.6363e-01,  1.9509e+00,  ..., -1.3107e+00,\n",
       "           -2.5669e-02, -2.6219e-01],\n",
       "          [-1.7174e+00,  4.1371e-01,  1.6879e-01,  ...,  3.3504e+00,\n",
       "           -2.4638e+00, -9.2596e-01],\n",
       "          [ 9.7647e-02,  5.0778e-01,  1.2721e+00,  ..., -7.3185e-01,\n",
       "            2.9798e-01,  7.3979e-01]],\n",
       "\n",
       "         [[-1.1668e+00, -9.8979e-01, -1.3641e+00,  ..., -1.6647e+00,\n",
       "           -2.8051e-01, -1.1136e+00],\n",
       "          [-1.0737e+00, -6.5732e-01, -4.5897e-01,  ..., -7.6946e-01,\n",
       "           -1.5687e-02, -1.2671e+00],\n",
       "          [-9.3979e-02,  1.0303e+00, -5.5281e-01,  ..., -4.5288e-01,\n",
       "           -1.6225e-01, -2.5431e+00],\n",
       "          ...,\n",
       "          [ 1.7823e+00,  3.7671e-01, -1.3733e+00,  ...,  1.6706e+00,\n",
       "            1.7361e+00, -1.8360e+00],\n",
       "          [-5.9223e-01, -1.9264e+00,  4.4903e-01,  ...,  4.9339e-01,\n",
       "           -1.7304e-01, -5.2651e-01],\n",
       "          [-1.0624e+00, -9.4940e-01,  3.4107e-01,  ...,  6.0012e-01,\n",
       "           -1.0600e+00, -9.2872e-01]],\n",
       "\n",
       "         [[ 9.5309e-02,  5.3553e-01,  7.7978e-02,  ..., -2.0109e-02,\n",
       "            5.1027e-01,  1.0304e+00],\n",
       "          [-5.9817e-01, -6.2534e-01,  5.9577e-02,  ..., -1.0416e+00,\n",
       "            2.4246e-01,  1.4550e+00],\n",
       "          [ 8.7974e-02, -5.2563e-01,  7.7894e-01,  ...,  1.1555e+00,\n",
       "            1.2669e+00, -7.6856e-02],\n",
       "          ...,\n",
       "          [-1.3301e+00,  2.1888e-02,  2.2022e-01,  ..., -1.1289e+00,\n",
       "           -6.1456e-01, -2.6682e-01],\n",
       "          [-2.5832e+00, -2.9969e+00, -3.1691e+00,  ..., -2.6410e+00,\n",
       "           -1.5158e+00,  4.1706e+00],\n",
       "          [-3.6775e-01,  5.5610e-02,  8.9014e-02,  ..., -1.0719e+00,\n",
       "           -1.0172e+00,  1.2555e+00]]],\n",
       "\n",
       "\n",
       "        [[[-8.8532e-01, -5.4065e-01, -1.7415e+00,  ...,  2.7014e-01,\n",
       "           -8.7790e-01, -6.1585e-01],\n",
       "          [-2.0049e-01,  8.2937e-01,  3.5621e-01,  ..., -7.5599e-01,\n",
       "            1.0060e+00, -5.7967e-01],\n",
       "          [ 2.0313e+00,  9.4101e-01, -1.1603e+00,  ..., -6.1833e-02,\n",
       "           -3.7763e-01, -2.9338e-01],\n",
       "          ...,\n",
       "          [-7.1895e-02,  1.0393e+00, -1.6934e+00,  ..., -1.2618e-01,\n",
       "            7.9219e-01, -1.2131e+00],\n",
       "          [-9.3615e-01,  2.8850e+00,  2.0324e+00,  ..., -2.2315e+00,\n",
       "            1.5525e+00,  2.5596e+00],\n",
       "          [ 7.0311e-02, -5.0812e-01,  1.2361e-01,  ...,  3.6916e-01,\n",
       "            4.5180e-01,  2.0020e-01]],\n",
       "\n",
       "         [[-5.0733e-01,  5.2527e-02,  7.9587e-01,  ...,  7.0143e-01,\n",
       "           -1.1795e-01,  6.1197e-01],\n",
       "          [-6.4586e-01,  3.8522e-01, -2.8933e-01,  ...,  7.1054e-01,\n",
       "            5.9605e-01,  4.2540e-01],\n",
       "          [ 7.5557e-03,  3.5966e-01,  1.0600e+00,  ..., -8.4839e-01,\n",
       "           -1.2277e+00,  7.4038e-01],\n",
       "          ...,\n",
       "          [ 9.6443e-01,  6.6435e-01, -2.5579e-01,  ..., -4.3202e-01,\n",
       "           -7.2114e-01, -1.2826e-01],\n",
       "          [-1.7924e+00, -1.8184e+00, -9.1941e-01,  ...,  9.1197e-01,\n",
       "           -4.3646e-01, -8.9395e-01],\n",
       "          [-7.5563e-01,  4.9015e-02,  7.4589e-02,  ...,  1.9563e+00,\n",
       "            1.0951e+00, -5.1333e-01]],\n",
       "\n",
       "         [[ 1.1277e+00, -2.0251e+00, -4.5678e-01,  ..., -6.7217e-01,\n",
       "            1.8660e+00, -1.5452e+00],\n",
       "          [-2.4965e+00,  7.2907e-01,  2.0456e-01,  ...,  2.5770e+00,\n",
       "           -4.1919e-01,  2.3096e+00],\n",
       "          [-1.5538e+00,  1.0435e+00, -4.1812e-01,  ...,  3.9723e-01,\n",
       "           -7.8722e-01,  1.9145e+00],\n",
       "          ...,\n",
       "          [-1.6926e-01,  1.3546e+00,  2.2819e-01,  ...,  6.2981e-01,\n",
       "           -1.8747e+00,  1.0251e+00],\n",
       "          [ 2.7004e+00, -3.2580e+00, -2.9331e+00,  ..., -2.3478e+00,\n",
       "            5.3549e-01, -2.8643e+00],\n",
       "          [-1.4112e-01, -9.9378e-02, -4.2941e-01,  ...,  1.7746e+00,\n",
       "           -3.2590e-01, -5.5224e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.8642e-01,  6.1583e-01, -6.6632e-01,  ..., -4.5940e-01,\n",
       "           -1.4822e+00,  1.2653e+00],\n",
       "          [-4.9021e-01,  1.8267e+00, -2.0262e+00,  ...,  2.3427e-01,\n",
       "           -1.5570e+00,  7.1884e-01],\n",
       "          [ 1.4636e+00, -6.0770e-02, -2.4201e+00,  ..., -1.8155e+00,\n",
       "            6.8077e-03, -4.7903e-01],\n",
       "          ...,\n",
       "          [ 7.8012e-01, -4.7551e-02,  1.2812e-01,  ..., -8.3930e-01,\n",
       "           -3.6147e-01,  4.1384e-01],\n",
       "          [-1.8064e+00,  4.7020e-01, -3.2840e-02,  ...,  3.3316e+00,\n",
       "           -2.5492e+00, -1.0371e+00],\n",
       "          [ 1.8722e-01, -3.9587e-01,  7.8315e-01,  ...,  5.5245e-01,\n",
       "           -5.6351e-01,  6.2563e-01]],\n",
       "\n",
       "         [[-1.1868e+00, -9.9299e-01, -1.3791e+00,  ..., -1.6761e+00,\n",
       "           -2.7219e-01, -1.1059e+00],\n",
       "          [-1.0975e+00, -6.8051e-01, -4.5312e-01,  ..., -7.8072e-01,\n",
       "            1.7720e-02, -1.2790e+00],\n",
       "          [ 1.5130e-01,  2.1704e+00, -7.5634e-01,  ..., -4.8734e-01,\n",
       "            1.3768e+00, -1.4434e+00],\n",
       "          ...,\n",
       "          [ 1.0288e+00,  1.0950e+00,  4.6369e-01,  ...,  1.6742e+00,\n",
       "           -1.0775e+00,  5.6829e-01],\n",
       "          [-5.8629e-01, -1.9080e+00,  4.6555e-01,  ...,  5.2090e-01,\n",
       "           -1.2238e-01, -5.5652e-01],\n",
       "          [-4.7717e-01, -1.5189e+00,  4.5009e-01,  ...,  1.0563e+00,\n",
       "           -2.6616e-01, -7.2594e-02]],\n",
       "\n",
       "         [[ 1.1273e-01,  5.3711e-01,  7.0958e-02,  ..., -1.3583e-02,\n",
       "            5.0991e-01,  1.0238e+00],\n",
       "          [-6.0528e-01, -6.8343e-01,  5.8132e-02,  ..., -1.0372e+00,\n",
       "            2.2856e-01,  1.4431e+00],\n",
       "          [ 2.2147e+00,  8.4521e-01,  1.3050e-01,  ...,  6.5358e-01,\n",
       "           -6.4170e-01, -6.6089e-01],\n",
       "          ...,\n",
       "          [-5.1802e-01, -4.6016e-01, -1.4261e+00,  ..., -3.0014e-04,\n",
       "           -2.6083e-01, -8.3753e-01],\n",
       "          [-2.8471e+00, -3.1420e+00, -3.4942e+00,  ..., -2.7838e+00,\n",
       "           -1.7751e+00,  4.3338e+00],\n",
       "          [-9.1022e-01, -9.0150e-01, -6.5721e-01,  ..., -1.9500e+00,\n",
       "           -9.6951e-01,  1.1626e+00]]],\n",
       "\n",
       "\n",
       "        [[[-8.9642e-01, -5.2744e-01, -1.7365e+00,  ...,  2.6663e-01,\n",
       "           -8.7320e-01, -6.2661e-01],\n",
       "          [-2.5577e-01,  8.5166e-01,  3.3627e-01,  ..., -7.0525e-01,\n",
       "            9.5653e-01, -5.9257e-01],\n",
       "          [-4.6450e-01,  7.5495e-01,  1.5739e-01,  ...,  1.3400e-01,\n",
       "           -1.0938e-02, -1.8761e-01],\n",
       "          ...,\n",
       "          [ 2.1057e+00, -1.7279e-01, -6.7083e-01,  ...,  2.0100e-01,\n",
       "            9.0114e-01, -8.2595e-02],\n",
       "          [-2.9761e-01, -3.8496e-01,  4.9501e-01,  ..., -1.0624e+00,\n",
       "            6.8003e-01,  3.6499e-01],\n",
       "          [ 3.0436e-01,  2.0438e-02, -2.3921e-01,  ..., -1.2220e+00,\n",
       "            5.7388e-01,  4.1578e-01]],\n",
       "\n",
       "         [[-5.0645e-01,  5.3403e-02,  7.9692e-01,  ...,  7.0396e-01,\n",
       "           -1.0178e-01,  6.0466e-01],\n",
       "          [-6.5703e-01,  3.7151e-01, -2.8487e-01,  ...,  7.2263e-01,\n",
       "            6.1317e-01,  4.2151e-01],\n",
       "          [-9.1934e-01,  5.3902e-01, -3.4656e-01,  ...,  1.3705e+00,\n",
       "            3.3404e-01, -1.1501e+00],\n",
       "          ...,\n",
       "          [-7.7485e-02,  2.5777e-01,  1.2772e-01,  ..., -6.6947e-01,\n",
       "            2.3217e-01, -1.8527e-01],\n",
       "          [-8.6978e-01, -6.3627e-01, -1.4441e+00,  ..., -1.7225e-01,\n",
       "            1.1242e+00, -7.7492e-02],\n",
       "          [ 1.8071e+00, -5.7783e-02, -4.3027e-01,  ...,  5.3235e-01,\n",
       "            2.9343e-02, -1.3988e-01]],\n",
       "\n",
       "         [[ 1.1234e+00, -2.0217e+00, -4.6270e-01,  ..., -6.9496e-01,\n",
       "            1.8547e+00, -1.5345e+00],\n",
       "          [-2.4962e+00,  7.4567e-01,  1.6445e-01,  ...,  2.4982e+00,\n",
       "           -4.5251e-01,  2.3106e+00],\n",
       "          [ 7.3729e-01, -1.6438e-01,  2.5789e-02,  ...,  7.4082e-01,\n",
       "           -6.3472e-01,  8.3509e-01],\n",
       "          ...,\n",
       "          [-1.5329e+00,  2.2617e+00,  2.4375e-01,  ...,  4.3600e-01,\n",
       "           -1.1299e-01,  6.7529e-01],\n",
       "          [ 1.0737e+00, -8.4013e-01, -5.0385e-01,  ...,  1.0421e+00,\n",
       "           -3.3810e-02,  1.5827e-01],\n",
       "          [ 2.1016e-02,  1.2225e+00,  3.5818e-01,  ...,  8.0039e-01,\n",
       "            4.8317e-01, -1.9103e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.9395e-01,  5.9849e-01, -6.5415e-01,  ..., -4.4767e-01,\n",
       "           -1.4938e+00,  1.2705e+00],\n",
       "          [-5.0567e-01,  1.7826e+00, -1.9740e+00,  ...,  2.5127e-01,\n",
       "           -1.5837e+00,  7.5272e-01],\n",
       "          [ 6.1877e-01,  1.2552e-01, -1.9401e-01,  ..., -2.1986e-01,\n",
       "           -9.2903e-01,  6.6197e-01],\n",
       "          ...,\n",
       "          [ 3.4379e-01,  6.1926e-01,  9.1491e-02,  ..., -1.1301e+00,\n",
       "           -8.6144e-01,  1.7692e+00],\n",
       "          [ 1.2830e-01,  6.9542e-01, -1.7933e-01,  ..., -1.1849e-02,\n",
       "           -1.5170e+00,  1.8923e+00],\n",
       "          [ 1.2501e+00, -9.0351e-01,  1.9051e-01,  ..., -1.6533e+00,\n",
       "            1.1749e+00,  1.1116e+00]],\n",
       "\n",
       "         [[-1.1776e+00, -1.0059e+00, -1.3608e+00,  ..., -1.6748e+00,\n",
       "           -2.7358e-01, -1.0985e+00],\n",
       "          [-1.0955e+00, -7.0135e-01, -4.2019e-01,  ..., -7.4563e-01,\n",
       "           -1.7227e-02, -1.2465e+00],\n",
       "          [ 4.8129e-01,  1.5142e+00,  8.9042e-01,  ..., -5.9277e-01,\n",
       "            9.9060e-01,  2.5922e-01],\n",
       "          ...,\n",
       "          [ 6.9422e-01, -4.6429e-01,  1.6141e+00,  ...,  5.5032e-01,\n",
       "            1.5677e+00,  1.2588e+00],\n",
       "          [-1.2923e+00, -2.4530e+00,  1.1306e+00,  ...,  2.5075e-01,\n",
       "            5.9220e-01, -9.0370e-01],\n",
       "          [-4.0500e-01, -1.1689e+00,  7.3900e-01,  ...,  4.6328e-01,\n",
       "            7.7912e-01,  1.1154e+00]],\n",
       "\n",
       "         [[ 1.2155e-01,  5.4589e-01,  8.1009e-02,  ..., -2.9373e-02,\n",
       "            5.1307e-01,  1.0199e+00],\n",
       "          [-5.8590e-01, -6.2934e-01,  6.9415e-02,  ..., -1.0558e+00,\n",
       "            2.6424e-01,  1.4117e+00],\n",
       "          [ 2.8019e-01,  7.5664e-01,  1.6572e+00,  ..., -2.5114e-01,\n",
       "            2.8801e-01, -3.3370e-02],\n",
       "          ...,\n",
       "          [ 1.0138e+00,  9.2862e-01,  2.1937e-01,  ..., -4.8950e-01,\n",
       "           -7.7442e-01, -5.1576e-01],\n",
       "          [-3.4716e-01, -4.2840e-01, -7.3395e-02,  ..., -2.0734e+00,\n",
       "           -1.4484e-01,  5.8984e-01],\n",
       "          [ 8.9212e-01,  5.8408e-01, -6.7369e-01,  ..., -1.0548e+00,\n",
       "            3.2137e-01,  1.7269e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-4.4937e-01, -2.6759e-01,  2.3972e-01,  ..., -8.1488e-02,\n",
       "            6.0648e-01,  4.7460e-03],\n",
       "          [ 7.5429e-02, -8.6441e-01, -1.2596e-01,  ..., -4.9939e-02,\n",
       "            4.9605e-01,  2.0360e-01],\n",
       "          [-2.4440e+00,  1.1925e-01,  2.3047e+00,  ...,  8.1536e-01,\n",
       "           -4.3480e-02,  3.9932e-01],\n",
       "          ...,\n",
       "          [-1.2376e+00,  9.1264e-01,  1.2538e+00,  ..., -1.3706e+00,\n",
       "           -1.3262e-01,  7.5398e-01],\n",
       "          [ 6.3166e-02, -1.0123e+00, -8.7893e-01,  ..., -6.5656e-01,\n",
       "            3.4503e-02, -2.6909e-01],\n",
       "          [-5.0284e-02, -1.4033e+00, -1.4631e-01,  ...,  1.5106e+00,\n",
       "            3.8114e-02, -9.8489e-01]],\n",
       "\n",
       "         [[ 1.3576e-01,  1.0406e-01,  2.4032e-01,  ..., -4.7403e-01,\n",
       "            4.3627e-01, -5.4124e-01],\n",
       "          [ 9.0684e-01, -1.0553e-02, -5.2141e-02,  ...,  7.5999e-03,\n",
       "            3.5421e-01, -1.0607e-01],\n",
       "          [-1.2486e-01, -5.0206e-01,  6.8141e-01,  ..., -1.9072e-01,\n",
       "           -1.7918e-01,  8.5898e-01],\n",
       "          ...,\n",
       "          [ 1.6909e-01, -1.9071e-02, -1.2761e+00,  ..., -4.3385e-01,\n",
       "            1.2837e-02,  1.2852e+00],\n",
       "          [ 1.2327e+00, -2.2543e+00,  1.9867e+00,  ...,  2.4383e+00,\n",
       "           -3.3761e+00,  4.0902e+00],\n",
       "          [-3.6014e-01, -2.8245e-01, -3.3124e-01,  ..., -4.3714e-01,\n",
       "            5.6238e-01, -9.9477e-02]],\n",
       "\n",
       "         [[-2.0580e+00, -1.6560e+00, -1.9552e+00,  ...,  1.4805e+00,\n",
       "           -1.0013e+00, -5.6165e-01],\n",
       "          [ 4.1413e-01,  3.7762e-01, -2.6711e-02,  ...,  5.8588e-01,\n",
       "            5.4468e-02,  1.1715e-01],\n",
       "          [ 2.2026e+00, -1.7756e-01, -1.9345e+00,  ..., -4.3116e-01,\n",
       "           -1.8131e-01, -7.1247e-01],\n",
       "          ...,\n",
       "          [-7.7363e-01,  1.1593e+00,  9.3782e-01,  ..., -6.8743e-01,\n",
       "           -9.9665e-01, -2.0116e-02],\n",
       "          [ 1.2164e+00, -4.7849e-01,  1.4323e+00,  ...,  1.3166e+00,\n",
       "           -1.4639e-01,  5.0656e-01],\n",
       "          [ 3.0422e-01,  1.4911e+00,  1.6241e-01,  ...,  1.4238e-01,\n",
       "            3.2861e-01, -4.1136e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.7245e-01, -9.5249e-02,  3.0360e-01,  ...,  5.4966e-01,\n",
       "            3.3323e-01,  3.2985e-01],\n",
       "          [-2.9045e-01, -1.0141e-01, -1.9576e-01,  ..., -1.0389e-01,\n",
       "            2.9613e-01, -7.4658e-01],\n",
       "          [ 4.1620e-01,  9.4670e-01,  4.4945e-01,  ..., -2.6242e-01,\n",
       "            8.4691e-01,  5.0817e-01],\n",
       "          ...,\n",
       "          [-1.7911e+00,  2.9878e-02,  1.4467e+00,  ...,  1.2555e+00,\n",
       "            9.1164e-01,  1.6402e+00],\n",
       "          [ 4.0550e-01,  9.2819e-01, -9.7617e-01,  ...,  1.1337e+00,\n",
       "            2.4257e-01,  8.2743e-01],\n",
       "          [ 7.0771e-01,  7.0866e-01,  6.1033e-01,  ..., -9.5067e-01,\n",
       "           -9.4512e-01, -1.1532e+00]],\n",
       "\n",
       "         [[-7.3148e-02, -6.5462e-01, -4.5310e-02,  ..., -4.1860e-02,\n",
       "            7.0225e-01, -5.4711e-01],\n",
       "          [ 5.1215e-01, -5.6059e-01, -1.2699e-02,  ..., -3.0220e-01,\n",
       "            4.9039e-01, -5.6014e-02],\n",
       "          [-5.7377e-01, -3.1240e-01, -1.1184e-02,  ...,  4.4348e-01,\n",
       "            1.2963e+00, -3.8560e-01],\n",
       "          ...,\n",
       "          [-4.9566e-01, -4.8626e-02,  1.5843e-01,  ...,  1.5316e+00,\n",
       "           -1.1115e+00, -2.2606e-01],\n",
       "          [-8.5656e-01, -1.6188e+00,  3.7642e-01,  ..., -3.7529e-01,\n",
       "           -6.5104e-02, -7.5114e-01],\n",
       "          [ 1.8170e-02, -4.0168e-01, -8.5334e-01,  ...,  1.0808e-01,\n",
       "           -1.0492e+00,  2.1481e-04]],\n",
       "\n",
       "         [[ 6.4300e-01, -4.8592e-01, -5.7925e-02,  ..., -2.5264e-01,\n",
       "            4.4641e-01,  7.3707e-02],\n",
       "          [-3.5636e-01, -8.4747e-01, -3.2702e-01,  ...,  5.8737e-01,\n",
       "           -3.4949e-01,  1.8938e-01],\n",
       "          [-4.9487e-02, -9.0992e-01, -1.1898e+00,  ...,  2.8281e-01,\n",
       "            1.0348e+00,  1.4022e+00],\n",
       "          ...,\n",
       "          [ 2.6852e-01, -7.4539e-01, -3.1545e-01,  ...,  1.1890e-01,\n",
       "           -1.1697e-01, -2.4020e-01],\n",
       "          [ 3.4159e-01,  3.5713e-01,  4.6117e-01,  ..., -9.0582e-01,\n",
       "           -3.2063e-01,  1.6396e+00],\n",
       "          [-3.0646e-01, -5.2065e-01, -4.2979e-01,  ..., -1.7998e-01,\n",
       "            4.2892e-01,  8.5524e-01]]],\n",
       "\n",
       "\n",
       "        [[[-4.3845e-01, -2.8449e-01,  2.5384e-01,  ..., -7.0283e-02,\n",
       "            6.1816e-01,  8.5345e-03],\n",
       "          [ 1.2523e-01, -8.8518e-01, -1.0525e-01,  ..., -2.0424e-02,\n",
       "            5.4560e-01,  2.2826e-01],\n",
       "          [-6.8498e-01, -3.5626e-01,  4.9457e-01,  ...,  3.7831e-02,\n",
       "           -1.0008e+00,  1.8361e-01],\n",
       "          ...,\n",
       "          [ 3.7518e-01, -4.7903e-01,  1.1551e+00,  ..., -1.6202e+00,\n",
       "            1.1769e+00,  3.9505e-01],\n",
       "          [ 2.0579e-01, -1.1473e+00, -9.6956e-01,  ..., -6.9548e-01,\n",
       "            1.8239e-01, -2.6752e-01],\n",
       "          [-7.9661e-02, -1.4481e-01,  1.4401e-01,  ...,  3.9887e-01,\n",
       "            2.7084e-01, -3.7034e-01]],\n",
       "\n",
       "         [[ 1.1963e-01,  1.1795e-01,  2.4972e-01,  ..., -4.7860e-01,\n",
       "            4.3165e-01, -5.4390e-01],\n",
       "          [ 9.1915e-01,  7.3682e-02, -3.3234e-02,  ...,  4.3251e-02,\n",
       "            3.9091e-01, -1.2187e-01],\n",
       "          [ 1.3992e+00,  2.0111e-01, -3.9343e-02,  ...,  1.1727e+00,\n",
       "           -2.1005e+00,  5.1992e-01],\n",
       "          ...,\n",
       "          [ 1.1347e+00, -1.2315e+00, -8.1363e-01,  ...,  4.1176e-01,\n",
       "            3.3522e-01,  7.1366e-01],\n",
       "          [ 1.1988e+00, -2.2356e+00,  2.1357e+00,  ...,  2.4424e+00,\n",
       "           -3.4362e+00,  4.0948e+00],\n",
       "          [-6.0065e-01,  1.2357e-01, -3.5629e-02,  ..., -2.2997e-01,\n",
       "            8.2677e-02,  3.0730e-02]],\n",
       "\n",
       "         [[-2.0895e+00, -1.6731e+00, -1.9647e+00,  ...,  1.4859e+00,\n",
       "           -9.9986e-01, -5.5572e-01],\n",
       "          [ 3.6171e-01,  3.8619e-01, -3.9722e-03,  ...,  5.6685e-01,\n",
       "            6.9974e-02,  1.3924e-01],\n",
       "          [ 6.3401e-01,  2.3894e-01,  1.1109e+00,  ...,  1.3319e+00,\n",
       "            3.0544e-01,  2.0257e+00],\n",
       "          ...,\n",
       "          [ 5.3166e-01, -1.0387e+00,  1.9174e-01,  ...,  3.1501e-01,\n",
       "            1.1306e+00,  9.0549e-01],\n",
       "          [ 1.2155e+00, -4.9732e-01,  1.4142e+00,  ...,  1.2415e+00,\n",
       "           -2.1032e-01,  4.7529e-01],\n",
       "          [ 3.1437e-01,  8.8799e-01, -7.0295e-01,  ..., -5.1047e-01,\n",
       "            1.3139e-01,  2.2217e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.6882e-01, -1.0047e-01,  2.9638e-01,  ...,  5.2742e-01,\n",
       "            3.3515e-01,  3.3048e-01],\n",
       "          [-3.2197e-01, -1.2855e-01, -2.1894e-01,  ..., -1.7050e-01,\n",
       "            2.9843e-01, -7.0562e-01],\n",
       "          [ 7.2463e-01, -5.4304e-01, -1.3449e+00,  ...,  1.5530e+00,\n",
       "            1.4806e+00,  8.0724e-02],\n",
       "          ...,\n",
       "          [ 8.5239e-01, -1.2690e+00,  1.3924e+00,  ..., -9.3014e-02,\n",
       "            1.3505e+00, -5.2947e-01],\n",
       "          [ 3.8343e-01,  8.4282e-01, -1.1008e+00,  ...,  9.5561e-01,\n",
       "            2.7692e-01,  7.8251e-01],\n",
       "          [-9.3326e-01,  4.8113e-01,  2.3551e+00,  ..., -3.4650e-01,\n",
       "           -1.1965e-01, -5.7442e-01]],\n",
       "\n",
       "         [[-8.5915e-02, -6.6741e-01, -5.8828e-02,  ..., -4.9445e-02,\n",
       "            7.0619e-01, -5.6112e-01],\n",
       "          [ 4.2257e-01, -6.5221e-01, -3.1406e-02,  ..., -2.5923e-01,\n",
       "            4.3861e-01, -7.2715e-02],\n",
       "          [-2.5040e+00, -6.4272e-01,  5.7287e-01,  ..., -4.6420e-01,\n",
       "           -1.0010e+00,  2.3605e-01],\n",
       "          ...,\n",
       "          [-3.1746e-01, -4.6495e-02, -8.1118e-01,  ..., -5.5608e-02,\n",
       "            6.6925e-01, -3.4340e-01],\n",
       "          [-1.0046e+00, -1.7356e+00,  2.7791e-01,  ..., -3.7102e-01,\n",
       "           -1.2691e-01, -7.4941e-01],\n",
       "          [-1.4494e+00, -1.4095e+00, -8.1863e-01,  ..., -1.0421e+00,\n",
       "           -5.9359e-01, -3.0197e-01]],\n",
       "\n",
       "         [[ 6.4325e-01, -4.9444e-01, -5.8244e-02,  ..., -2.5394e-01,\n",
       "            4.4569e-01,  7.0682e-02],\n",
       "          [-3.5883e-01, -9.2083e-01, -3.5726e-01,  ...,  5.4033e-01,\n",
       "           -3.6818e-01,  1.6586e-01],\n",
       "          [ 8.1305e-01, -1.0482e+00,  2.6796e-01,  ..., -1.4673e+00,\n",
       "           -5.9731e-01,  1.0726e+00],\n",
       "          ...,\n",
       "          [ 3.9077e-01, -1.6804e+00,  2.6807e-01,  ...,  1.4888e+00,\n",
       "            2.2817e+00,  3.2351e+00],\n",
       "          [ 3.3012e-01,  3.2909e-01,  4.5043e-01,  ..., -9.3161e-01,\n",
       "           -3.9514e-01,  1.6884e+00],\n",
       "          [ 7.2230e-01, -8.6734e-01, -8.7342e-01,  ...,  1.0961e+00,\n",
       "           -1.8225e-01, -1.1480e-01]]],\n",
       "\n",
       "\n",
       "        [[[-4.4780e-01, -2.7175e-01,  2.3759e-01,  ..., -7.8359e-02,\n",
       "            6.1430e-01, -8.9904e-03],\n",
       "          [ 1.3486e-01, -8.7980e-01, -1.3994e-01,  ..., -1.8633e-02,\n",
       "            5.2830e-01,  1.7714e-01],\n",
       "          [ 7.9678e-02,  1.1158e+00, -2.2950e-01,  ..., -4.1124e-01,\n",
       "           -1.3391e+00, -8.8613e-01],\n",
       "          ...,\n",
       "          [ 8.8698e-02, -9.0519e-01,  3.9073e-01,  ..., -6.7586e-01,\n",
       "            1.1510e-01, -1.0828e-01],\n",
       "          [ 3.1180e-01,  1.5459e-01,  6.4565e-01,  ...,  5.0351e-01,\n",
       "           -4.6244e-01, -3.0517e-01],\n",
       "          [ 4.6112e-01, -2.5227e-01,  1.7358e-01,  ..., -1.9210e-01,\n",
       "           -7.9860e-02,  3.2413e-01]],\n",
       "\n",
       "         [[ 1.1644e-01,  1.0398e-01,  2.6024e-01,  ..., -4.8067e-01,\n",
       "            4.3306e-01, -5.4246e-01],\n",
       "          [ 8.6811e-01,  3.6325e-02, -2.8294e-02,  ...,  1.4371e-03,\n",
       "            4.0012e-01, -1.5586e-01],\n",
       "          [ 8.0668e-02,  1.1253e+00,  6.0176e-01,  ..., -4.2910e-01,\n",
       "           -1.0494e+00,  2.6624e-01],\n",
       "          ...,\n",
       "          [-4.4065e-01,  4.4606e-01, -9.8375e-01,  ...,  2.5354e-01,\n",
       "           -9.6376e-01,  3.6808e-01],\n",
       "          [-2.2348e-01, -5.0917e-01, -7.7683e-01,  ..., -6.7325e-01,\n",
       "           -2.6125e-01, -3.8594e-01],\n",
       "          [-1.1469e+00, -5.7863e-01, -1.2936e+00,  ..., -4.7592e-01,\n",
       "           -2.3709e-01,  6.5400e-01]],\n",
       "\n",
       "         [[-2.0996e+00, -1.6751e+00, -1.9581e+00,  ...,  1.4931e+00,\n",
       "           -1.0155e+00, -5.5816e-01],\n",
       "          [ 3.4776e-01,  3.5232e-01, -3.5032e-03,  ...,  6.1845e-01,\n",
       "            5.7270e-02,  9.5129e-02],\n",
       "          [ 3.9319e-01,  1.3991e-01, -9.2118e-01,  ...,  9.4202e-01,\n",
       "            9.9275e-03, -5.6658e-02],\n",
       "          ...,\n",
       "          [ 3.2836e-01,  1.0382e+00,  2.2875e+00,  ...,  1.0961e+00,\n",
       "           -4.0726e-01,  5.4125e-01],\n",
       "          [ 7.5758e-01, -9.5204e-02,  6.5999e-02,  ..., -1.1602e-01,\n",
       "           -3.1502e-01, -8.6526e-02],\n",
       "          [ 7.9733e-01, -1.5799e-01,  5.7341e-01,  ..., -2.3248e-01,\n",
       "           -1.9471e-01,  1.2299e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.7313e-01, -8.6664e-02,  2.9878e-01,  ...,  5.4265e-01,\n",
       "            3.3401e-01,  3.4414e-01],\n",
       "          [-2.9666e-01, -9.4124e-02, -2.0949e-01,  ..., -1.3388e-01,\n",
       "            3.2218e-01, -6.7733e-01],\n",
       "          [-1.7250e+00,  7.4973e-01,  7.7588e-01,  ...,  1.2573e+00,\n",
       "            1.1398e+00,  6.9919e-01],\n",
       "          ...,\n",
       "          [-8.0268e-01, -1.8515e+00, -9.9356e-01,  ...,  1.7265e+00,\n",
       "            1.7973e+00,  1.6007e-01],\n",
       "          [-1.2282e-01,  1.0070e+00,  1.3339e+00,  ...,  1.9542e-01,\n",
       "            1.1317e+00,  1.4473e-01],\n",
       "          [-3.3623e-01, -3.4963e-01,  9.5046e-01,  ...,  3.0605e-01,\n",
       "            4.3325e-01, -5.9113e-01]],\n",
       "\n",
       "         [[-7.0579e-02, -6.7623e-01, -4.4333e-02,  ..., -2.8728e-02,\n",
       "            7.0827e-01, -5.6274e-01],\n",
       "          [ 4.6432e-01, -6.6327e-01, -4.2066e-02,  ..., -2.0704e-01,\n",
       "            4.3629e-01, -7.8188e-02],\n",
       "          [-1.1821e+00, -1.8696e-01,  2.1491e-01,  ...,  1.5820e-01,\n",
       "           -1.3435e+00, -3.3055e-01],\n",
       "          ...,\n",
       "          [-1.0697e+00,  1.3637e+00, -7.4791e-01,  ..., -4.8116e-01,\n",
       "            9.4784e-01,  4.5608e-01],\n",
       "          [-1.0340e+00, -9.3239e-01,  7.4240e-03,  ...,  3.8345e-01,\n",
       "           -8.6700e-01, -2.0537e+00],\n",
       "          [-5.7121e-01,  3.9489e-01, -6.6367e-01,  ...,  5.8912e-01,\n",
       "            1.5459e+00,  7.0131e-01]],\n",
       "\n",
       "         [[ 6.4509e-01, -4.9227e-01, -5.4922e-02,  ..., -2.5218e-01,\n",
       "            4.4832e-01,  7.2878e-02],\n",
       "          [-3.5585e-01, -8.9555e-01, -3.4745e-01,  ...,  5.7335e-01,\n",
       "           -3.5922e-01,  1.5585e-01],\n",
       "          [-4.9566e-01, -7.7548e-01,  6.1178e-01,  ..., -2.1242e-01,\n",
       "           -7.7515e-01,  1.0129e+00],\n",
       "          ...,\n",
       "          [ 7.1044e-01, -1.2223e+00, -6.2687e-01,  ..., -5.0907e-01,\n",
       "            2.4895e-01,  1.1238e+00],\n",
       "          [ 2.5406e-01, -5.9567e-01, -7.9098e-01,  ...,  3.7142e-01,\n",
       "           -2.3967e-02,  1.3019e+00],\n",
       "          [ 1.3256e-01, -1.0970e+00, -1.6462e+00,  ...,  6.3656e-01,\n",
       "           -8.7016e-01, -1.0024e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 2.5249e+00, -1.3450e+00,  2.8183e+00,  ...,  1.4435e+00,\n",
       "           -1.6538e+00,  2.8359e-02],\n",
       "          [ 1.3908e+00, -1.1278e+00,  7.5779e-01,  ...,  1.1424e+00,\n",
       "           -1.6204e+00, -6.8631e-01],\n",
       "          [ 1.6840e+00, -1.1369e+00,  3.1363e+00,  ...,  1.3301e-01,\n",
       "           -1.4887e+00,  2.3601e-01],\n",
       "          ...,\n",
       "          [ 9.3256e-01, -7.2499e-01,  3.6567e+00,  ..., -7.9259e-01,\n",
       "           -2.3599e+00,  3.8693e-01],\n",
       "          [-2.4772e-01, -3.0027e+00,  1.9273e+00,  ..., -2.0123e-01,\n",
       "           -1.7498e+00, -3.4445e-01],\n",
       "          [ 4.1361e-01, -1.6537e+00, -3.4455e-01,  ...,  5.5741e-01,\n",
       "            3.1587e+00,  2.7747e-03]],\n",
       "\n",
       "         [[-1.9601e+00,  1.8484e+00,  6.2567e-01,  ..., -1.3849e+00,\n",
       "            1.7444e-01,  1.9493e+00],\n",
       "          [ 4.9735e-01, -9.2061e-01,  9.5699e-02,  ..., -7.7414e-01,\n",
       "           -4.4245e-01, -9.2320e-02],\n",
       "          [ 4.3832e-01, -5.7232e-01,  6.4520e-01,  ..., -2.1363e-01,\n",
       "            1.6414e+00, -2.6528e-02],\n",
       "          ...,\n",
       "          [ 1.0863e+00,  1.1163e+00, -5.5702e-02,  ...,  1.1775e-01,\n",
       "            3.3485e-01, -1.6714e+00],\n",
       "          [ 1.0484e+00,  6.4319e-01,  4.9860e-01,  ...,  1.1996e+00,\n",
       "            1.0157e+00,  1.1665e+00],\n",
       "          [-2.4977e+00,  1.0136e+00,  1.7654e-01,  ...,  1.9922e+00,\n",
       "            2.1563e-02, -1.1024e+00]],\n",
       "\n",
       "         [[ 1.2019e-01, -2.0332e-02,  6.8694e-02,  ..., -1.0953e+00,\n",
       "            1.9432e-01,  2.8023e+00],\n",
       "          [-2.3622e-01, -7.4958e-01, -7.5780e-01,  ..., -1.4991e+00,\n",
       "           -1.7967e+00,  1.2088e+00],\n",
       "          [-9.8768e-01,  1.3201e-01, -1.7319e-01,  ..., -4.6084e-02,\n",
       "           -1.5132e+00,  2.1519e+00],\n",
       "          ...,\n",
       "          [ 2.0111e+00, -3.8302e-01,  1.2396e+00,  ..., -1.1973e+00,\n",
       "           -2.3710e-01, -1.9327e-01],\n",
       "          [ 7.2789e-01, -8.1663e-01,  3.5312e-01,  ..., -2.9568e-01,\n",
       "           -2.3502e+00,  8.8297e-01],\n",
       "          [-1.3524e+00, -4.2844e-01, -2.7179e-01,  ...,  4.2081e-01,\n",
       "           -1.9083e+00, -9.8728e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.3358e+00,  2.5232e-01,  2.0226e+00,  ..., -2.9864e-01,\n",
       "           -1.5831e+00, -9.6747e-01],\n",
       "          [ 1.9274e+00, -3.6307e-01, -2.4351e+00,  ...,  4.4522e-01,\n",
       "            9.4490e-01,  1.1594e+00],\n",
       "          [ 1.5212e+00,  3.9019e-01, -4.3010e-01,  ...,  2.0637e+00,\n",
       "            1.6316e+00, -5.4483e-01],\n",
       "          ...,\n",
       "          [ 1.3603e+00,  1.0826e+00,  3.5446e-01,  ...,  7.4110e-01,\n",
       "           -3.3472e-01, -3.0448e+00],\n",
       "          [ 3.9513e+00,  1.1313e+00, -1.0318e+00,  ...,  1.1311e-01,\n",
       "            1.4538e+00,  2.3684e+00],\n",
       "          [ 1.0430e+00,  9.3369e-01, -5.4070e-01,  ..., -2.1946e+00,\n",
       "           -1.7381e+00,  2.6507e-01]],\n",
       "\n",
       "         [[ 4.5591e-01, -8.3729e-01, -1.4637e+00,  ..., -1.7491e+00,\n",
       "           -3.8516e+00,  1.1717e+00],\n",
       "          [-2.6594e-01, -6.8152e-01, -1.8961e+00,  ..., -1.8720e+00,\n",
       "           -1.4639e+00, -3.0592e-01],\n",
       "          [-1.2106e+00, -2.2451e+00, -1.4986e+00,  ..., -1.1971e+00,\n",
       "           -2.8926e-01, -6.9404e-01],\n",
       "          ...,\n",
       "          [-8.5142e-01, -9.8774e-01, -1.3279e+00,  ..., -5.9900e-01,\n",
       "           -1.6056e-01,  3.6634e-01],\n",
       "          [-1.6187e+00, -1.2129e+00, -1.1807e+00,  ...,  2.5491e-01,\n",
       "           -1.2375e+00, -3.2913e-01],\n",
       "          [ 1.2439e+00,  2.7216e+00,  7.4772e-01,  ..., -2.2026e+00,\n",
       "            1.6152e+00, -1.0420e+00]],\n",
       "\n",
       "         [[-5.8911e-01, -3.7743e-01, -7.1607e-01,  ...,  2.4395e+00,\n",
       "            2.9423e+00,  1.0787e+00],\n",
       "          [-2.4381e-01, -1.0157e+00, -3.5702e-01,  ...,  1.4354e+00,\n",
       "           -2.4606e-01, -2.0882e+00],\n",
       "          [ 1.6194e+00, -2.1923e+00, -1.4783e+00,  ..., -1.7098e+00,\n",
       "            6.2679e-01, -8.4773e-01],\n",
       "          ...,\n",
       "          [ 1.7723e+00,  5.2650e-01,  4.9519e-01,  ...,  5.2709e-01,\n",
       "           -4.0345e-01, -2.6967e+00],\n",
       "          [ 9.0959e-01, -9.0214e-01, -3.4420e+00,  ..., -2.8570e+00,\n",
       "           -1.0846e+00, -6.1383e-01],\n",
       "          [-2.0796e+00,  2.9173e+00,  1.8560e-01,  ...,  1.9171e+00,\n",
       "            2.6154e+00,  2.5403e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 2.1569e+00, -2.2402e+00,  2.7141e+00,  ...,  6.6912e-01,\n",
       "           -8.0250e-01,  1.7160e+00],\n",
       "          [ 1.2727e+00, -1.2522e+00,  3.3917e+00,  ..., -5.6814e-01,\n",
       "           -1.6761e+00,  1.6796e+00],\n",
       "          [ 2.4126e-01, -3.9022e-01,  2.6140e+00,  ..., -1.2303e+00,\n",
       "            8.3344e-01, -7.3105e-02],\n",
       "          ...,\n",
       "          [ 2.6836e+00, -2.1402e+00,  2.3860e+00,  ...,  8.3485e-01,\n",
       "           -7.7107e-01,  1.0250e+00],\n",
       "          [ 2.5238e-01, -3.1256e+00, -2.9436e-01,  ...,  1.6876e+00,\n",
       "           -1.3834e+00, -6.5227e-01],\n",
       "          [ 4.2524e-01, -1.6366e+00, -3.8261e-01,  ...,  5.9637e-01,\n",
       "            3.1817e+00, -1.2552e-02]],\n",
       "\n",
       "         [[ 3.8960e-01,  2.1304e+00,  1.0106e+00,  ..., -1.1082e+00,\n",
       "           -8.3367e-01,  7.7327e-01],\n",
       "          [ 4.6482e-01,  5.0898e-01,  5.4163e-01,  ...,  2.4406e-01,\n",
       "            2.3451e-01, -1.2818e+00],\n",
       "          [ 1.8056e+00, -6.6286e-01,  1.0521e+00,  ...,  9.0721e-01,\n",
       "           -2.7160e-01, -3.4135e-02],\n",
       "          ...,\n",
       "          [-4.7730e-01,  1.0096e+00,  5.0987e-01,  ...,  5.0037e-01,\n",
       "           -4.7615e-01, -1.6124e+00],\n",
       "          [-1.9064e+00,  6.4916e-01,  1.0267e+00,  ...,  3.5189e-01,\n",
       "           -4.3736e-01,  1.0292e+00],\n",
       "          [-2.5490e+00,  1.0057e+00,  2.1141e-01,  ...,  1.9702e+00,\n",
       "            2.2580e-02, -1.1233e+00]],\n",
       "\n",
       "         [[-3.0623e-01, -1.5690e+00, -9.7545e-01,  ..., -1.4781e+00,\n",
       "            1.6084e-01,  1.6368e+00],\n",
       "          [-5.2070e-01,  2.9465e-01, -5.9236e-02,  ..., -1.0374e+00,\n",
       "           -1.2372e-01,  2.4883e+00],\n",
       "          [-5.3093e-01,  1.4546e-01, -1.2253e+00,  ..., -1.5917e+00,\n",
       "            1.7489e+00,  1.0229e+00],\n",
       "          ...,\n",
       "          [ 3.5938e-01, -2.4837e+00, -2.2342e+00,  ..., -6.1377e-01,\n",
       "           -1.0262e+00,  1.8597e+00],\n",
       "          [ 1.4095e+00, -5.1534e-01, -1.8254e+00,  ...,  2.2141e-01,\n",
       "           -3.1365e+00,  2.8600e-01],\n",
       "          [-1.3617e+00, -4.2235e-01, -3.7493e-01,  ...,  4.8258e-01,\n",
       "           -1.9288e+00, -1.0427e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 3.8484e-01, -3.4603e+00,  1.9580e-01,  ...,  1.7745e+00,\n",
       "           -1.6538e+00, -4.4736e-01],\n",
       "          [ 1.1105e+00,  2.7620e-01,  9.9783e-01,  ...,  1.8886e+00,\n",
       "            1.7839e+00, -1.2407e-01],\n",
       "          [ 5.1273e-01, -2.2911e+00, -1.6810e+00,  ...,  1.7753e+00,\n",
       "            1.9086e+00,  4.5876e-01],\n",
       "          ...,\n",
       "          [ 8.3614e-01,  1.2206e+00, -1.3706e+00,  ..., -6.9934e-01,\n",
       "           -9.1336e-01, -2.0249e+00],\n",
       "          [ 3.4242e+00,  1.6732e+00, -1.7963e-01,  ..., -1.5798e+00,\n",
       "            7.2017e-01,  3.1292e+00],\n",
       "          [ 1.0931e+00,  9.5625e-01, -5.3024e-01,  ..., -2.2244e+00,\n",
       "           -1.7780e+00,  2.4507e-01]],\n",
       "\n",
       "         [[ 5.5184e-02, -1.3463e+00,  1.0560e+00,  ..., -1.2943e+00,\n",
       "           -1.1129e+00,  1.1011e+00],\n",
       "          [-7.8981e-01, -2.4461e+00, -1.1616e+00,  ..., -6.3598e-01,\n",
       "           -1.1390e+00,  3.9426e-01],\n",
       "          [ 1.6430e-01, -1.3176e+00,  1.9239e-01,  ..., -4.3497e-01,\n",
       "           -2.6126e-01,  3.0478e-01],\n",
       "          ...,\n",
       "          [-9.7611e-01, -5.5258e-01, -4.2884e-01,  ..., -3.0951e-01,\n",
       "           -1.6480e+00, -1.3182e+00],\n",
       "          [-9.1629e-01, -1.4000e-01, -1.4554e+00,  ...,  8.4655e-01,\n",
       "           -2.0292e+00,  7.2846e-02],\n",
       "          [ 1.2397e+00,  2.7655e+00,  7.4571e-01,  ..., -2.1882e+00,\n",
       "            1.6117e+00, -1.0277e+00]],\n",
       "\n",
       "         [[-4.5307e-01, -1.8624e+00,  5.9248e-01,  ...,  9.5919e-01,\n",
       "            2.3591e+00, -1.7583e+00],\n",
       "          [ 1.1376e+00, -8.5284e-01, -1.7664e+00,  ..., -1.7706e+00,\n",
       "           -1.2188e-01, -1.0192e+00],\n",
       "          [ 5.0067e-01, -1.2487e+00,  2.7835e-01,  ...,  5.0180e-01,\n",
       "            3.0839e-01,  1.2730e-01],\n",
       "          ...,\n",
       "          [ 2.4172e-01, -1.0506e+00,  3.5384e-01,  ...,  9.2819e-01,\n",
       "            1.0919e+00,  1.7782e-01],\n",
       "          [ 6.8456e-01, -1.1087e+00, -3.5520e+00,  ..., -2.9044e+00,\n",
       "           -5.9810e-01,  6.7364e-02],\n",
       "          [-2.1243e+00,  2.9587e+00,  1.6366e-01,  ...,  1.9270e+00,\n",
       "            2.5903e+00,  2.5835e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 7.6790e-01, -2.1101e-01,  4.6367e+00,  ...,  1.5097e+00,\n",
       "           -8.0609e-01, -8.4278e-01],\n",
       "          [-7.1600e-01,  5.3171e-01,  4.9983e+00,  ...,  2.0680e+00,\n",
       "            2.3431e-01,  4.8930e-01],\n",
       "          [-5.2424e-01,  2.0138e+00,  3.4945e+00,  ...,  2.9408e+00,\n",
       "           -1.8648e+00, -3.0759e+00],\n",
       "          ...,\n",
       "          [-1.0194e+00,  1.3671e+00,  3.4354e-01,  ..., -1.1262e-01,\n",
       "           -6.8467e-01, -1.7485e+00],\n",
       "          [-7.3106e-01, -1.9734e+00,  2.4392e+00,  ...,  4.1649e-01,\n",
       "            5.6757e-02, -9.2306e-01],\n",
       "          [ 3.9338e-01, -1.6014e+00, -3.7576e-01,  ...,  6.0964e-01,\n",
       "            3.1776e+00,  3.1903e-02]],\n",
       "\n",
       "         [[ 1.0006e+00,  9.6583e-01, -2.5487e-01,  ...,  8.4657e-01,\n",
       "            1.4511e+00, -1.7247e+00],\n",
       "          [ 1.0390e+00,  1.6249e+00,  1.1400e+00,  ..., -1.5005e-01,\n",
       "            2.4036e+00, -1.0465e+00],\n",
       "          [ 7.1793e-01,  6.6943e-01, -2.4826e-01,  ...,  1.1291e+00,\n",
       "            2.0425e+00, -3.6208e+00],\n",
       "          ...,\n",
       "          [-9.3486e-01, -7.8535e-01, -4.7990e-02,  ...,  1.9375e+00,\n",
       "            2.5909e+00,  3.8943e-04],\n",
       "          [ 5.0936e-01, -3.6325e-01,  1.4050e+00,  ...,  8.6911e-01,\n",
       "            9.0762e-01,  1.1079e+00],\n",
       "          [-2.6090e+00,  1.0123e+00,  2.0519e-01,  ...,  2.0397e+00,\n",
       "            3.9270e-02, -1.1281e+00]],\n",
       "\n",
       "         [[-1.2531e-01,  2.4467e-01, -1.0928e+00,  ..., -3.0995e+00,\n",
       "            8.5216e-01,  1.1625e+00],\n",
       "          [-1.9582e+00, -4.9657e-01,  1.7196e-01,  ..., -1.5735e+00,\n",
       "            4.8576e-01,  2.0687e+00],\n",
       "          [ 1.3676e+00,  1.3117e+00, -1.4346e+00,  ..., -2.0302e+00,\n",
       "           -8.8826e-01,  9.1876e-01],\n",
       "          ...,\n",
       "          [ 2.2263e+00,  4.2958e-01, -2.2040e-01,  ..., -1.1207e+00,\n",
       "            1.0167e+00,  2.0616e+00],\n",
       "          [ 1.7370e+00, -1.6030e+00, -2.2887e-01,  ..., -4.8910e-01,\n",
       "           -1.7925e+00,  1.0077e+00],\n",
       "          [-1.3536e+00, -4.2548e-01, -3.7118e-01,  ...,  4.1570e-01,\n",
       "           -1.9003e+00, -9.8409e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.9604e-01, -1.2640e+00, -5.1219e-01,  ..., -3.0359e-01,\n",
       "            1.5025e+00, -8.3346e-02],\n",
       "          [ 2.1762e+00, -2.0148e+00,  1.0233e+00,  ...,  5.4225e-01,\n",
       "            6.0849e-01, -9.5647e-01],\n",
       "          [ 5.8989e-01, -1.2962e+00,  7.0052e-01,  ..., -1.5374e-02,\n",
       "            8.4950e-01, -1.4915e-01],\n",
       "          ...,\n",
       "          [ 1.3158e+00,  2.6017e-01,  1.2557e+00,  ..., -5.3860e-01,\n",
       "           -4.0121e-01, -5.5807e-01],\n",
       "          [ 3.1549e+00,  3.9858e-02, -2.8249e-01,  ..., -1.2880e+00,\n",
       "            1.2771e+00,  2.5422e+00],\n",
       "          [ 1.0041e+00,  9.2147e-01, -5.4803e-01,  ..., -2.2435e+00,\n",
       "           -1.7363e+00,  2.9854e-01]],\n",
       "\n",
       "         [[ 1.8749e+00, -7.8136e-01,  6.1805e-01,  ...,  1.9806e-01,\n",
       "            7.2301e-01,  1.1751e+00],\n",
       "          [ 1.1117e+00,  9.1694e-01,  1.4992e+00,  ...,  1.0367e+00,\n",
       "           -2.9042e-01,  1.5609e+00],\n",
       "          [ 1.4003e+00, -1.4346e+00,  1.3939e+00,  ..., -5.9248e-01,\n",
       "            6.3402e-01,  2.7954e+00],\n",
       "          ...,\n",
       "          [-1.7583e+00, -9.6793e-01, -6.8262e-01,  ..., -5.2564e-01,\n",
       "           -1.5691e-01,  8.6526e-01],\n",
       "          [-2.5025e+00,  4.6021e-01, -7.9392e-01,  ...,  8.0501e-01,\n",
       "           -7.1648e-01,  1.1753e+00],\n",
       "          [ 1.2910e+00,  2.7645e+00,  7.7386e-01,  ..., -2.1711e+00,\n",
       "            1.6444e+00, -9.9022e-01]],\n",
       "\n",
       "         [[-7.2865e-01, -4.3107e-01, -9.4584e-01,  ...,  2.7710e+00,\n",
       "            1.7315e+00,  3.9600e-01],\n",
       "          [ 1.1752e+00,  1.7378e-01, -5.1608e-01,  ...,  4.1172e-01,\n",
       "            8.7419e-01, -1.0116e+00],\n",
       "          [ 1.5799e+00, -7.5537e-01, -1.5912e+00,  ..., -1.7960e+00,\n",
       "           -9.8324e-01, -2.4919e+00],\n",
       "          ...,\n",
       "          [ 1.4217e+00, -1.3327e+00, -3.9171e-01,  ..., -2.1726e+00,\n",
       "            8.0012e-01, -1.3226e+00],\n",
       "          [ 7.9394e-01, -7.8354e-02, -3.4725e+00,  ..., -2.5262e+00,\n",
       "            2.9460e-01, -1.1131e+00],\n",
       "          [-2.1566e+00,  2.8908e+00,  1.6944e-01,  ...,  1.9325e+00,\n",
       "            2.6492e+00,  2.5532e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.0321e-01, -7.0572e-01, -1.1598e+00,  ..., -7.1561e-02,\n",
       "            8.8381e-01,  4.9372e-02],\n",
       "          [ 2.7741e-02, -5.4246e-01, -1.2600e+00,  ...,  7.5296e-01,\n",
       "           -3.4324e+00,  2.5309e+00],\n",
       "          [ 2.7450e-01, -1.3964e+00, -1.1767e+00,  ...,  6.6070e-01,\n",
       "           -1.7419e+00,  2.0077e+00],\n",
       "          ...,\n",
       "          [ 2.3050e-01, -3.9449e-02,  4.1316e-01,  ...,  1.8010e+00,\n",
       "            6.2558e-02,  1.0524e-01],\n",
       "          [ 8.4915e-01, -1.0723e+00,  6.4925e-01,  ..., -1.5701e-01,\n",
       "            6.1697e-01,  1.6715e-01],\n",
       "          [ 4.9657e-01, -4.6600e-01,  5.4770e-01,  ..., -5.3386e-02,\n",
       "           -1.0239e+00, -2.5228e-01]],\n",
       "\n",
       "         [[-8.1254e-01, -1.4359e-01, -1.6920e+00,  ..., -6.8903e-01,\n",
       "           -4.4195e-01, -1.5743e+00],\n",
       "          [ 5.7684e-02,  3.4646e-01, -2.4548e+00,  ..., -3.7670e-01,\n",
       "            1.9349e-01, -6.5194e-01],\n",
       "          [ 2.0284e-01,  4.0685e-01,  3.4692e-02,  ...,  1.0857e+00,\n",
       "           -3.0027e-01,  7.0187e-01],\n",
       "          ...,\n",
       "          [-1.9563e+00,  1.6056e+00, -1.0222e+00,  ...,  7.4398e-01,\n",
       "           -9.4049e-01,  3.8737e-01],\n",
       "          [-6.7319e-01, -4.2516e-01,  9.6486e-01,  ...,  8.0457e-01,\n",
       "           -8.1625e-01,  1.1011e+00],\n",
       "          [ 5.9756e-01, -1.5563e-01, -2.4473e-01,  ...,  4.2504e-01,\n",
       "           -2.1261e-01, -4.4851e-01]],\n",
       "\n",
       "         [[ 9.1703e-01,  9.3683e-01,  1.9833e-02,  ..., -2.3485e-01,\n",
       "            2.0722e+00,  1.4814e+00],\n",
       "          [ 1.9848e-02,  1.8733e+00, -4.3680e-01,  ...,  7.0849e-01,\n",
       "            1.4849e+00, -1.5488e-01],\n",
       "          [ 3.0273e-01,  1.6203e+00, -2.8364e-02,  ...,  2.1360e-01,\n",
       "           -3.8582e-01,  2.8078e-01],\n",
       "          ...,\n",
       "          [-5.4744e-02,  1.8465e+00, -1.9778e+00,  ...,  8.9542e-01,\n",
       "            1.0145e+00,  1.7239e+00],\n",
       "          [-7.0800e-01,  1.1638e-01, -6.0407e-01,  ...,  9.6518e-01,\n",
       "            9.1415e-01, -3.5849e-01],\n",
       "          [ 3.6570e-01, -1.7798e-01,  9.6035e-02,  ...,  3.3990e-01,\n",
       "            2.4154e-01,  1.0471e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.4698e+00,  1.0826e+00, -2.5936e+00,  ...,  2.4113e-01,\n",
       "           -1.4947e+00, -3.0658e+00],\n",
       "          [ 1.8606e+00, -1.8252e+00, -5.9825e-01,  ..., -6.0453e+00,\n",
       "           -4.2819e+00, -1.5650e+00],\n",
       "          [ 2.6652e+00,  9.2919e-02, -6.1882e-01,  ..., -5.0509e+00,\n",
       "           -3.1714e+00, -3.9297e-01],\n",
       "          ...,\n",
       "          [-9.6500e-02, -1.2317e+00, -6.1927e-01,  ..., -2.7859e+00,\n",
       "           -7.6453e-01, -2.9074e+00],\n",
       "          [-7.7752e-01,  2.5340e-02,  3.4603e+00,  ..., -5.4394e-01,\n",
       "            8.8783e-01, -2.3312e+00],\n",
       "          [-4.6493e-01, -2.5207e-01, -7.0410e-02,  ...,  1.3276e-01,\n",
       "            9.5329e-02, -2.9312e-01]],\n",
       "\n",
       "         [[-8.3338e-01,  1.8765e-01,  4.2380e-01,  ..., -5.8820e-03,\n",
       "            9.0065e-01, -1.4559e-01],\n",
       "          [-3.4494e-01, -6.7033e-01, -8.0665e-01,  ..., -1.6865e-01,\n",
       "           -6.6728e-01,  1.4793e-01],\n",
       "          [-4.6583e-01, -7.1144e-01,  7.7911e-01,  ...,  8.6542e-01,\n",
       "            1.9613e-01,  2.1406e-01],\n",
       "          ...,\n",
       "          [ 3.6975e-01, -9.8026e-01, -3.6081e-01,  ..., -2.4445e-01,\n",
       "            1.5641e-01, -3.1042e-01],\n",
       "          [ 6.3770e-01, -1.5896e+00,  3.8168e-01,  ...,  3.2062e-01,\n",
       "            9.1077e-01,  5.1709e-01],\n",
       "          [-9.7104e-02,  8.9912e-01,  2.5841e-01,  ...,  6.8359e-01,\n",
       "            3.7860e-02,  1.3172e-01]],\n",
       "\n",
       "         [[-2.6464e+00,  9.5897e-01, -9.5464e-01,  ...,  7.5237e-01,\n",
       "           -1.3941e+00,  6.0660e-01],\n",
       "          [ 2.0743e+00,  2.8842e+00,  4.2390e+00,  ...,  1.9113e+00,\n",
       "           -1.6381e+00,  2.5948e+00],\n",
       "          [-4.9345e-01,  1.3297e+00,  1.3210e+00,  ...,  3.9935e-01,\n",
       "           -3.2454e-01, -2.8012e+00],\n",
       "          ...,\n",
       "          [-1.6313e+00, -3.0922e+00,  4.3565e+00,  ..., -5.1549e-01,\n",
       "           -5.1387e-01,  1.0170e+00],\n",
       "          [ 2.7341e+00,  1.6857e+00, -1.4710e+00,  ..., -1.1746e+00,\n",
       "            8.0675e-01, -3.1815e+00],\n",
       "          [-1.8283e-01,  1.0846e-01, -3.4503e-02,  ..., -4.3794e-02,\n",
       "            1.0269e-01, -3.2443e-01]]],\n",
       "\n",
       "\n",
       "        [[[-4.5580e-01, -8.9008e-01,  3.9848e-01,  ...,  5.3784e-01,\n",
       "            1.9840e-01, -1.1002e+00],\n",
       "          [ 3.6262e-01, -4.2024e-01, -1.0572e+00,  ...,  7.0643e-01,\n",
       "           -1.7508e-01,  3.5968e-01],\n",
       "          [ 2.5479e+00, -1.6965e+00,  9.1315e-01,  ...,  1.0965e+00,\n",
       "            1.5069e-01, -8.9336e-01],\n",
       "          ...,\n",
       "          [-9.0986e-01,  5.8219e-01,  5.1346e-01,  ...,  7.5725e-01,\n",
       "            1.6316e-01,  2.4691e+00],\n",
       "          [ 2.5778e-02, -5.1711e-01, -4.1212e-01,  ..., -4.1365e-01,\n",
       "            1.3137e-01,  5.2798e-01],\n",
       "          [ 4.9843e-01, -4.9935e-01,  4.9806e-01,  ..., -1.0387e-01,\n",
       "           -9.9965e-01, -2.4763e-01]],\n",
       "\n",
       "         [[-6.9527e-01, -4.2079e-01, -5.0477e-01,  ...,  1.0643e+00,\n",
       "           -1.5351e-01, -2.2204e-01],\n",
       "          [-5.9473e-01,  7.0976e-02,  8.6555e-01,  ...,  1.2508e+00,\n",
       "           -6.0568e-02, -5.3541e-02],\n",
       "          [-1.0446e+00, -9.6563e-01,  1.8912e+00,  ...,  1.3953e-01,\n",
       "            1.1419e+00,  1.4613e-01],\n",
       "          ...,\n",
       "          [ 6.1197e-02,  6.8241e-04, -1.0529e+00,  ..., -4.2149e-01,\n",
       "           -6.0950e-01, -1.7343e-01],\n",
       "          [-5.2304e-01, -7.0735e-01,  2.8523e-01,  ...,  1.2563e+00,\n",
       "           -8.7633e-01,  1.3774e-02],\n",
       "          [ 6.1984e-01, -1.1975e-01, -2.6410e-01,  ...,  4.3514e-01,\n",
       "           -2.5642e-01, -4.3399e-01]],\n",
       "\n",
       "         [[ 6.9194e-01,  1.4139e+00, -8.5107e-01,  ..., -1.1565e+00,\n",
       "            2.0394e+00,  2.4493e+00],\n",
       "          [ 3.7174e-01,  1.7946e+00, -1.4056e+00,  ..., -1.1074e+00,\n",
       "           -1.1298e+00,  1.0616e+00],\n",
       "          [ 1.6767e+00,  4.4809e-01, -4.1193e-01,  ...,  5.2607e-01,\n",
       "            1.1290e-01,  5.9748e-01],\n",
       "          ...,\n",
       "          [ 1.6846e+00,  8.1422e-01, -8.6884e-01,  ...,  7.5299e-01,\n",
       "            1.4346e+00,  2.0573e-01],\n",
       "          [-1.2198e+00, -1.0369e-01,  9.0570e-01,  ...,  1.0430e+00,\n",
       "            6.7062e-01, -8.6810e-01],\n",
       "          [ 3.5034e-01, -1.8109e-01,  1.5186e-01,  ...,  3.3083e-01,\n",
       "            2.1137e-01,  1.3426e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.6856e+00, -2.3676e+00,  8.6639e-01,  ..., -1.4479e-01,\n",
       "           -1.4647e+00, -1.0800e+00],\n",
       "          [-3.1195e-01, -8.1889e-01,  1.6983e+00,  ...,  3.3836e-01,\n",
       "           -1.2109e+00, -1.2986e+00],\n",
       "          [ 1.7614e-01,  2.6950e+00,  1.7572e+00,  ..., -3.9850e+00,\n",
       "           -4.3753e-01,  1.6048e+00],\n",
       "          ...,\n",
       "          [-3.8285e-01, -1.0681e+00, -8.5861e-01,  ..., -4.1524e-01,\n",
       "           -1.4975e-02, -4.0573e+00],\n",
       "          [ 6.3229e-01,  9.4443e-01,  6.8037e-01,  ..., -5.8316e-01,\n",
       "            7.6143e-01, -5.1164e+00],\n",
       "          [-4.1374e-01, -3.2604e-01, -2.2312e-01,  ...,  1.3370e-01,\n",
       "            1.4275e-01, -5.0446e-01]],\n",
       "\n",
       "         [[-7.3138e-01, -3.4019e-02, -1.6668e+00,  ..., -1.2850e+00,\n",
       "            7.9229e-01, -1.6561e+00],\n",
       "          [-1.0332e-01, -2.1275e+00,  8.2209e-01,  ...,  4.0209e-02,\n",
       "            8.1043e-01,  3.5090e-02],\n",
       "          [ 5.0770e-01, -1.4846e+00,  1.0658e+00,  ..., -1.1770e+00,\n",
       "            4.8999e-01, -4.0947e-01],\n",
       "          ...,\n",
       "          [-3.0062e-01, -9.7482e-02,  5.9399e-01,  ..., -6.2674e-01,\n",
       "            2.9599e-01, -1.8088e+00],\n",
       "          [-7.1063e-01,  2.2734e-01,  1.6130e-01,  ...,  1.3451e+00,\n",
       "            4.5151e-01, -9.6677e-01],\n",
       "          [-9.5253e-02,  9.1124e-01,  2.5594e-01,  ...,  7.1123e-01,\n",
       "            6.4406e-03,  1.3949e-01]],\n",
       "\n",
       "         [[-3.2396e+00,  3.1091e-01,  3.1729e+00,  ...,  2.8876e-01,\n",
       "           -6.8117e-01,  4.4394e-01],\n",
       "          [-6.6483e-01, -2.5467e-01,  7.9327e-01,  ..., -1.1971e+00,\n",
       "            8.2032e-01, -4.4249e+00],\n",
       "          [ 4.5563e+00,  1.1836e+00, -4.3324e+00,  ...,  1.8670e+00,\n",
       "           -5.4413e+00, -1.4838e+00],\n",
       "          ...,\n",
       "          [ 7.8195e-01,  1.5208e-01,  5.7381e-01,  ...,  2.6145e+00,\n",
       "           -2.6231e+00, -1.4968e+00],\n",
       "          [-2.7331e+00, -3.2588e-02, -3.9994e-01,  ..., -2.2740e+00,\n",
       "           -6.3978e-01,  1.2496e+00],\n",
       "          [-2.7414e-01,  9.5921e-02,  1.1478e-01,  ..., -3.7485e-02,\n",
       "            7.5852e-02, -4.2951e-01]]],\n",
       "\n",
       "\n",
       "        [[[-4.7910e-01,  1.9337e-02,  7.3294e-01,  ...,  2.3050e-01,\n",
       "           -9.3336e-01, -5.1753e-01],\n",
       "          [ 7.7479e-01, -6.9933e-01,  6.7294e-01,  ...,  2.5810e-02,\n",
       "            7.4994e-01, -5.6137e-01],\n",
       "          [ 3.7078e-01,  6.4853e-02, -9.3768e-02,  ...,  9.2392e-01,\n",
       "            4.7467e-01,  2.9079e-01],\n",
       "          ...,\n",
       "          [ 1.0096e-01, -1.0085e-01, -9.0245e-01,  ...,  2.4677e-02,\n",
       "           -9.0234e-01, -5.7762e-01],\n",
       "          [ 4.7949e-01,  8.7144e-01, -1.5377e-01,  ..., -6.5088e-01,\n",
       "            5.0340e-01, -1.4913e+00],\n",
       "          [ 4.7473e-01, -4.8648e-01,  5.0636e-01,  ..., -1.0031e-01,\n",
       "           -1.0089e+00, -3.0043e-01]],\n",
       "\n",
       "         [[-4.9780e-01,  2.4897e-01, -1.6962e+00,  ..., -5.4082e-02,\n",
       "            2.6120e-01, -1.8201e+00],\n",
       "          [-3.1443e-01, -9.4778e-01, -7.1632e-01,  ...,  1.9061e+00,\n",
       "            5.1824e-01, -1.7728e-01],\n",
       "          [ 1.6805e+00, -8.0987e-01, -2.7621e+00,  ..., -7.2030e-01,\n",
       "           -1.2075e+00, -3.8705e-01],\n",
       "          ...,\n",
       "          [-1.7904e-01, -1.8553e+00,  1.4312e-01,  ..., -4.6090e-01,\n",
       "           -4.0583e+00,  1.6860e+00],\n",
       "          [-2.6335e-01,  6.1066e-01, -4.6761e-01,  ...,  3.0600e-01,\n",
       "            1.1841e-01, -2.7577e-01],\n",
       "          [ 6.2501e-01, -1.4922e-01, -2.5340e-01,  ...,  4.5745e-01,\n",
       "           -2.5382e-01, -4.5553e-01]],\n",
       "\n",
       "         [[ 9.9295e-01,  3.9429e-01,  1.0464e+00,  ..., -4.1931e-01,\n",
       "            1.6063e+00,  1.1686e-01],\n",
       "          [ 2.0422e+00,  1.0341e+00,  1.4525e-02,  ...,  1.2448e+00,\n",
       "            5.2846e-01,  4.8535e-01],\n",
       "          [ 2.9262e-01,  5.1421e-01, -8.9338e-01,  ..., -1.2820e+00,\n",
       "            1.3595e-01,  2.2064e-01],\n",
       "          ...,\n",
       "          [-9.8782e-01, -1.3553e+00,  7.2819e-01,  ...,  2.8424e-01,\n",
       "            8.8680e-01,  1.3475e+00],\n",
       "          [-1.0806e-01, -8.4652e-01, -7.9949e-01,  ...,  4.3728e-01,\n",
       "            1.9731e+00,  5.7069e-01],\n",
       "          [ 3.8241e-01, -1.7548e-01,  1.2134e-01,  ...,  3.1642e-01,\n",
       "            2.2814e-01,  1.4123e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 6.2004e-01, -1.4732e+00, -1.1710e+00,  ...,  2.5327e+00,\n",
       "            3.3605e+00, -2.4495e+00],\n",
       "          [ 9.4186e-01,  2.2490e+00,  9.1978e-01,  ..., -5.7796e-01,\n",
       "            4.5947e+00,  1.0575e+00],\n",
       "          [ 7.5798e-01,  1.9194e+00, -1.9581e+00,  ...,  1.2975e-01,\n",
       "            5.9325e+00,  8.4400e-01],\n",
       "          ...,\n",
       "          [ 1.3851e+00, -1.5020e+00, -5.1706e-01,  ...,  1.1721e+00,\n",
       "            3.6862e+00, -2.2900e+00],\n",
       "          [ 1.5518e-01,  4.8231e-01, -7.6816e-02,  ...,  3.5243e+00,\n",
       "            2.7498e+00, -9.8959e-01],\n",
       "          [-4.8145e-01, -2.6987e-01, -2.0006e-01,  ...,  3.2520e-01,\n",
       "            2.1808e-01, -3.1459e-01]],\n",
       "\n",
       "         [[-4.4021e-01,  2.8640e-01, -2.4250e+00,  ...,  2.1084e-01,\n",
       "            4.2765e-01, -7.7456e-01],\n",
       "          [-9.7099e-01, -6.5467e-02,  3.9854e-01,  ...,  1.2244e+00,\n",
       "           -2.2449e-01,  1.5826e+00],\n",
       "          [-8.1625e-01,  1.6319e+00,  1.0394e-01,  ...,  1.1103e+00,\n",
       "            3.3988e-01, -1.7676e+00],\n",
       "          ...,\n",
       "          [-1.7983e+00,  9.1786e-01,  4.6955e-01,  ...,  1.4336e+00,\n",
       "           -3.0341e-01, -8.5838e-01],\n",
       "          [-1.1823e-01,  1.7956e-01,  8.6820e-01,  ...,  2.7984e-01,\n",
       "            2.4258e-01, -1.0971e+00],\n",
       "          [-1.1139e-01,  9.2198e-01,  2.6538e-01,  ...,  7.0471e-01,\n",
       "            3.3896e-02,  1.7226e-01]],\n",
       "\n",
       "         [[-1.9322e+00,  8.9347e-01,  7.4771e-01,  ...,  1.1274e+00,\n",
       "           -6.1435e-01,  4.4023e-01],\n",
       "          [-7.6121e-01, -9.4006e-01,  3.2761e-01,  ...,  1.8063e+00,\n",
       "            2.1597e+00, -9.1864e-01],\n",
       "          [-2.1597e+00, -1.4665e+00,  3.1393e+00,  ..., -3.9898e-01,\n",
       "           -5.7989e+00, -1.9785e+00],\n",
       "          ...,\n",
       "          [-2.4708e+00, -2.3867e+00,  5.2672e-01,  ..., -1.5986e+00,\n",
       "            1.0053e+00, -1.4253e+00],\n",
       "          [-2.1940e+00,  1.4192e+00, -2.9636e-01,  ...,  1.0227e+00,\n",
       "           -5.6958e-01, -3.6836e+00],\n",
       "          [-3.0797e-01,  1.1946e-01,  1.1823e-01,  ...,  3.9102e-02,\n",
       "            1.4234e-01, -2.7233e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-0.4732, -1.6652, -0.2511,  ...,  1.3769,  2.3977, -1.6563],\n",
       "          [ 0.0760, -0.7602,  1.0203,  ...,  0.1250, -0.4127, -0.6751],\n",
       "          [ 1.2705,  0.8585,  0.2876,  ...,  0.3507, -0.3198,  1.0248],\n",
       "          ...,\n",
       "          [-2.3081, -1.8638, -1.9857,  ..., -1.7543,  0.3102, -2.4180],\n",
       "          [-1.9521, -3.2522,  0.1335,  ..., -1.5353,  1.4608, -3.4874],\n",
       "          [-1.9994, -2.9282, -0.4102,  ...,  0.1295,  1.1554, -1.7524]],\n",
       "\n",
       "         [[-2.0031, -0.8541, -0.2981,  ..., -0.5382,  0.1951,  0.8072],\n",
       "          [ 0.9305,  0.0838, -1.2000,  ..., -1.0419,  1.1184,  0.9153],\n",
       "          [ 0.5744, -0.3026, -1.6586,  ...,  0.2898,  0.2532,  1.9448],\n",
       "          ...,\n",
       "          [-1.7578, -0.7237,  1.3220,  ..., -3.0418,  1.0602,  0.6753],\n",
       "          [-1.8498, -0.0160,  1.2979,  ..., -2.4300,  0.1006, -0.2641],\n",
       "          [-1.5090,  0.6047,  0.6424,  ..., -2.1863, -0.0310,  0.4854]],\n",
       "\n",
       "         [[-0.1770,  0.0685,  0.7529,  ...,  0.4504,  1.6101, -1.6903],\n",
       "          [-1.2787,  0.7187,  0.1738,  ...,  0.5576,  0.0806, -1.5805],\n",
       "          [-1.1864,  1.4300, -0.7942,  ...,  0.8000,  0.2926, -1.5739],\n",
       "          ...,\n",
       "          [ 1.3902, -1.0519,  0.6303,  ..., -0.3689,  1.5569, -0.1154],\n",
       "          [ 0.9475, -0.4528,  0.4080,  ..., -0.0203,  2.8452,  1.1625],\n",
       "          [-0.8539,  1.6684, -0.2620,  ..., -0.0958,  1.1609,  0.9510]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.4993, -1.4669,  1.5708,  ...,  3.3939, -0.3907,  0.9102],\n",
       "          [-0.8696,  0.9890,  0.2311,  ...,  0.9283, -1.3846, -0.5319],\n",
       "          [-0.7682,  0.4106, -0.2857,  ...,  0.7537, -2.0045, -1.4688],\n",
       "          ...,\n",
       "          [ 0.5565, -0.1437,  0.2143,  ..., -0.0154, -0.7795, -0.3024],\n",
       "          [ 1.1090, -1.2002,  3.1912,  ...,  1.8582, -1.1143,  0.8815],\n",
       "          [ 1.2693,  1.1423,  3.4504,  ...,  3.0818, -1.8485,  0.9356]],\n",
       "\n",
       "         [[ 0.7599, -0.3614, -0.4252,  ..., -0.8709,  0.1219,  0.1709],\n",
       "          [-0.7615,  0.1147,  0.5212,  ..., -0.5822,  1.4949, -1.2079],\n",
       "          [-0.6636,  1.1220,  0.0646,  ...,  0.1529, -0.2910,  1.6574],\n",
       "          ...,\n",
       "          [ 1.2951,  0.1586, -2.8803,  ..., -3.1161, -1.1701, -0.4370],\n",
       "          [ 2.2346,  0.1001, -3.6498,  ..., -2.6563,  0.5730, -2.4694],\n",
       "          [ 2.0918, -0.5709, -3.4828,  ..., -0.9567, -2.4301,  0.4894]],\n",
       "\n",
       "         [[ 2.3913, -0.8866, -1.8491,  ...,  2.8980,  0.9499,  0.0248],\n",
       "          [-0.6381,  0.2124, -0.2374,  ...,  1.0366,  0.0525,  1.4040],\n",
       "          [-0.1648, -0.6020,  0.5652,  ...,  0.2025, -0.0945,  1.3179],\n",
       "          ...,\n",
       "          [ 0.7544, -0.4402, -0.5370,  ..., -0.8714,  0.6315, -1.1073],\n",
       "          [ 1.2915, -1.1106, -1.5079,  ..., -0.8409,  1.9385, -0.9085],\n",
       "          [ 1.8268, -2.7935, -0.3605,  ...,  1.7530,  0.5494, -0.9496]]],\n",
       "\n",
       "\n",
       "        [[[-0.4910, -1.6625, -0.2054,  ...,  1.3526,  2.4241, -1.6084],\n",
       "          [-0.3855, -2.5675,  0.9873,  ...,  0.0763,  0.5842, -1.0377],\n",
       "          [ 1.1170, -0.3737, -0.2528,  ..., -0.7211, -1.3073,  0.2792],\n",
       "          ...,\n",
       "          [-1.4745, -0.8858,  0.5766,  ..., -1.0989,  0.2880, -1.6946],\n",
       "          [-1.9552, -1.6673,  0.0403,  ..., -1.5878,  1.6122, -2.3162],\n",
       "          [-2.4447, -3.1689,  0.2759,  ..., -1.1988,  1.2017, -2.0817]],\n",
       "\n",
       "         [[-2.0404, -0.8817, -0.2862,  ..., -0.5000,  0.1789,  0.7679],\n",
       "          [-0.6423, -0.4532, -0.3860,  ..., -0.2946, -0.3094,  1.3355],\n",
       "          [-0.4739, -0.2218,  0.3256,  ..., -0.0744, -0.5769,  2.6927],\n",
       "          ...,\n",
       "          [-1.8532, -1.7119,  1.1688,  ..., -1.3542,  0.4391,  0.8520],\n",
       "          [-1.7229, -0.2283, -0.3277,  ..., -2.6153,  1.3658,  0.4937],\n",
       "          [-3.1977,  0.7446,  1.0160,  ..., -2.0503,  0.5137,  0.4678]],\n",
       "\n",
       "         [[-0.1997,  0.1118,  0.7479,  ...,  0.3989,  1.6155, -1.7062],\n",
       "          [-1.4798,  1.4522,  0.7191,  ..., -0.1433, -0.0472, -1.6206],\n",
       "          [-2.3322,  1.6220,  1.0219,  ..., -0.6070,  0.6522, -1.5514],\n",
       "          ...,\n",
       "          [-0.4747,  0.6083,  0.2752,  ..., -0.8280,  2.1263, -0.1143],\n",
       "          [ 2.4671, -0.9019, -0.0192,  ...,  0.3478,  2.0763, -0.1969],\n",
       "          [-0.1244,  1.2189, -0.0417,  ..., -0.0882,  1.7239,  0.4546]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.5119, -1.4200,  1.5454,  ...,  3.3972, -0.3763,  0.9285],\n",
       "          [-1.4961, -2.0089,  0.0201,  ...,  2.9867, -0.4238, -0.2820],\n",
       "          [-0.5860, -2.1308, -0.5736,  ...,  0.8614, -2.3761, -0.7730],\n",
       "          ...,\n",
       "          [-1.7424, -0.9338,  0.7874,  ...,  0.0348, -0.7579, -0.1818],\n",
       "          [ 0.6444,  0.6726,  3.0941,  ...,  1.5565, -1.7001,  0.9489],\n",
       "          [ 1.8661, -0.7256,  3.4903,  ...,  2.8042, -1.4856,  1.0810]],\n",
       "\n",
       "         [[ 0.7875, -0.3711, -0.4236,  ..., -0.8679,  0.1327,  0.1905],\n",
       "          [-0.3541, -0.7068, -0.6206,  ..., -1.0334,  1.1684,  0.0314],\n",
       "          [ 0.0414,  0.5440, -0.3792,  ...,  1.0472, -0.3460,  0.9401],\n",
       "          ...,\n",
       "          [ 0.8519,  0.5874, -2.0207,  ..., -1.3956, -0.4816,  0.6376],\n",
       "          [ 1.0736,  0.2712, -3.2711,  ..., -1.6882, -0.9250, -2.1613],\n",
       "          [ 2.8656,  0.3539, -4.6460,  ..., -2.3357, -3.0036, -0.3875]],\n",
       "\n",
       "         [[ 2.4683, -0.8660, -1.8346,  ...,  2.9063,  0.9629,  0.0354],\n",
       "          [-0.4246,  0.2165,  0.9661,  ...,  1.1430, -0.6153, -0.0099],\n",
       "          [-0.0405, -1.3538,  0.4158,  ...,  1.7288, -0.9387,  0.7647],\n",
       "          ...,\n",
       "          [ 0.5556, -1.2031, -1.0641,  ...,  0.1958,  0.8473, -0.9332],\n",
       "          [ 0.9250, -0.6220, -1.8171,  ...,  0.0789,  1.3940,  0.1173],\n",
       "          [ 2.0114, -1.5492, -1.2585,  ...,  0.8311,  1.8573, -0.6466]]],\n",
       "\n",
       "\n",
       "        [[[-0.4735, -1.6640, -0.2218,  ...,  1.3762,  2.4116, -1.6086],\n",
       "          [ 0.1506, -1.5644,  0.7441,  ...,  0.1071, -1.0942, -0.3998],\n",
       "          [-0.6890, -0.7649,  0.6686,  ..., -1.0383, -0.6245, -0.8152],\n",
       "          ...,\n",
       "          [-0.6135, -2.0741, -0.6361,  ..., -0.6309, -0.7277, -2.0513],\n",
       "          [-1.5970, -2.9949,  0.3862,  ..., -1.5229,  1.0638, -2.4673],\n",
       "          [-2.4574, -2.8848, -0.8127,  ..., -0.4556,  0.4763, -2.2763]],\n",
       "\n",
       "         [[-1.9992, -0.8532, -0.2940,  ..., -0.4956,  0.2044,  0.7922],\n",
       "          [ 0.1009, -0.3597,  0.2967,  ..., -1.0012, -0.5236,  1.9587],\n",
       "          [ 0.0369, -1.2645,  0.7085,  ..., -0.1063, -0.6368,  3.0087],\n",
       "          ...,\n",
       "          [-2.7785, -1.0633,  2.3706,  ..., -2.4394,  0.2306,  2.2328],\n",
       "          [-2.7930, -0.3338,  1.1197,  ..., -2.0314,  1.0273,  0.6069],\n",
       "          [-2.8348,  1.0602,  1.1433,  ..., -2.7756,  0.1500,  0.9916]],\n",
       "\n",
       "         [[-0.1888,  0.1048,  0.7564,  ...,  0.4282,  1.6051, -1.6723],\n",
       "          [-1.2655,  0.4001,  1.3000,  ..., -0.9641,  1.1459, -0.5619],\n",
       "          [-0.9790,  1.1189,  0.6827,  ...,  0.8451, -0.3751, -1.9100],\n",
       "          ...,\n",
       "          [ 0.1561, -0.3868,  0.7992,  ..., -1.0006,  2.0445,  0.5411],\n",
       "          [ 2.8166, -0.8723, -0.5355,  ...,  0.9359,  2.4282,  0.2611],\n",
       "          [-1.1615,  0.3330, -0.3195,  ...,  0.4839,  1.9112,  1.9596]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.4976, -1.4443,  1.5226,  ...,  3.4030, -0.3646,  0.9147],\n",
       "          [-0.6741, -1.6430,  0.4395,  ...,  1.7051, -0.5465, -0.2829],\n",
       "          [-0.6383, -1.7567,  1.2035,  ...,  1.1268, -0.6685, -0.0719],\n",
       "          ...,\n",
       "          [ 2.0344, -1.8735,  1.0570,  ...,  2.0253, -1.0396,  1.8260],\n",
       "          [ 0.9551, -1.6217,  3.3751,  ...,  1.2186, -1.0362,  1.6816],\n",
       "          [ 2.5166,  0.2744,  3.0065,  ...,  3.3239, -1.0607,  0.8608]],\n",
       "\n",
       "         [[ 0.7663, -0.3861, -0.3811,  ..., -0.8357,  0.1317,  0.2010],\n",
       "          [ 1.4376, -0.0292, -1.5381,  ..., -0.6210,  1.2783, -0.2846],\n",
       "          [ 0.4514, -0.2719, -1.9176,  ..., -0.1499, -0.4193, -0.5144],\n",
       "          ...,\n",
       "          [ 3.9110,  0.4666, -2.7167,  ..., -1.5758,  0.4445, -1.5285],\n",
       "          [ 3.4454, -0.2527, -4.6959,  ..., -3.8819, -2.1414, -2.4173],\n",
       "          [ 3.7320,  0.6225, -3.4576,  ..., -1.2329, -2.1393,  0.1921]],\n",
       "\n",
       "         [[ 2.4152, -0.8713, -1.8357,  ...,  2.8962,  0.9240,  0.0529],\n",
       "          [-1.1674, -0.9452,  0.1023,  ...,  1.7716,  0.4189,  0.2463],\n",
       "          [-0.9218, -1.2516, -0.1487,  ...,  1.7397, -0.7817,  0.2128],\n",
       "          ...,\n",
       "          [ 0.2973, -1.3558, -0.8216,  ...,  0.6805,  1.9114,  0.2779],\n",
       "          [ 1.8990, -1.3645, -1.8278,  ..., -0.1118,  1.3426,  0.6189],\n",
       "          [ 1.8924, -2.1457, -1.9094,  ...,  0.7615,  0.9265, -0.3257]]]],\n",
       "       grad_fn=<TransposeBackward0>), tensor([[[[-6.6054e-01,  5.6224e-02, -2.1224e-01,  ..., -2.7761e-01,\n",
       "            1.9687e-01, -4.1268e-02],\n",
       "          [-1.0560e+00, -1.1799e+00,  6.7855e-01,  ..., -8.5382e-01,\n",
       "            1.5909e+00, -1.1573e+00],\n",
       "          [-9.0114e-01, -1.9600e+00,  7.7220e-01,  ..., -1.8842e-01,\n",
       "            1.4289e+00,  2.4356e-01],\n",
       "          ...,\n",
       "          [-7.5769e-01,  3.6987e-01, -1.6364e+00,  ..., -6.9325e-01,\n",
       "            7.6535e-01, -3.7867e-01],\n",
       "          [-2.4102e+00,  1.8402e+00, -1.2348e+00,  ..., -1.9816e-01,\n",
       "            8.2838e-01,  2.1825e-01],\n",
       "          [-1.2650e+00, -1.9992e+00, -1.4427e+00,  ...,  5.4320e-01,\n",
       "            1.2111e+00,  3.1724e-01]],\n",
       "\n",
       "         [[ 3.5890e-01,  1.0393e+00,  1.1822e-01,  ...,  6.3941e-01,\n",
       "           -6.4576e-02, -8.5207e-02],\n",
       "          [-3.6680e-01,  1.1793e+00, -6.3691e-01,  ..., -3.8436e-01,\n",
       "            5.6465e-01,  2.7184e-01],\n",
       "          [ 4.2343e-02,  8.2559e-01, -3.8290e-01,  ..., -1.1241e+00,\n",
       "            7.9912e-01, -7.8107e-01],\n",
       "          ...,\n",
       "          [ 7.1663e-02, -1.8032e-01, -4.8851e-02,  ...,  1.2588e+00,\n",
       "           -4.7805e-01,  2.4197e-01],\n",
       "          [ 9.7981e-01,  2.6439e-01,  1.4079e+00,  ..., -5.0733e-02,\n",
       "           -1.6659e-01, -1.2487e+00],\n",
       "          [-2.5631e-01,  5.2368e-01, -4.7285e-01,  ..., -1.6355e-01,\n",
       "           -1.9701e+00, -1.5646e+00]],\n",
       "\n",
       "         [[ 7.7600e-02,  1.4784e-01, -1.3178e-01,  ...,  2.8898e-01,\n",
       "            4.1929e-01,  1.1603e+00],\n",
       "          [ 2.2546e-01, -2.5592e-01, -8.9388e-01,  ..., -1.4139e-01,\n",
       "            8.3700e-01, -8.2517e-02],\n",
       "          [ 8.8077e-01, -2.5787e-01, -3.3908e-01,  ..., -4.0229e-01,\n",
       "            7.0223e-01,  7.2161e-02],\n",
       "          ...,\n",
       "          [ 1.6367e+00, -3.0091e-01, -9.3960e-01,  ..., -7.1818e-01,\n",
       "            7.5200e-01,  1.4827e+00],\n",
       "          [ 9.3605e-01, -5.0624e-02, -6.3732e-03,  ..., -1.7762e-01,\n",
       "           -3.6478e-01,  2.2012e-01],\n",
       "          [ 1.4772e-01, -2.3781e+00, -8.7677e-01,  ...,  9.2413e-01,\n",
       "            2.5547e+00,  4.1863e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.7474e-01,  2.8867e-01, -1.6164e-01,  ...,  1.0618e-01,\n",
       "            1.5652e-01,  1.0195e-01],\n",
       "          [ 5.1299e-01,  2.6843e-01, -2.4524e-01,  ..., -1.8199e-01,\n",
       "            1.4455e-01,  5.1247e-01],\n",
       "          [-1.6401e-01,  3.0783e-01, -5.7060e-04,  ..., -1.4641e-01,\n",
       "            5.8299e-01, -7.1891e-01],\n",
       "          ...,\n",
       "          [ 8.5249e-02, -9.5167e-01, -1.2406e-01,  ...,  5.1373e-01,\n",
       "            6.0761e-02, -3.0958e-01],\n",
       "          [-3.5486e-01,  9.6092e-01, -3.2567e-01,  ...,  8.9373e-01,\n",
       "            1.1931e-01,  1.5778e+00],\n",
       "          [ 2.1719e+00,  6.2612e-01,  1.1643e+00,  ..., -9.0931e-01,\n",
       "           -2.0907e+00, -3.0086e-01]],\n",
       "\n",
       "         [[-1.4940e+00,  3.3551e-01, -1.1396e-01,  ...,  6.4738e-01,\n",
       "            5.2148e-01,  1.3671e+00],\n",
       "          [-2.8190e-01, -3.8178e-01, -7.1867e-01,  ...,  1.1523e+00,\n",
       "            2.0814e+00,  1.4094e+00],\n",
       "          [ 2.7481e-01,  4.9037e-01, -1.4817e+00,  ..., -1.1077e-01,\n",
       "            2.6348e+00, -2.0689e-01],\n",
       "          ...,\n",
       "          [ 1.0112e-01,  3.6266e-01, -1.8794e-02,  ..., -4.5541e-01,\n",
       "            2.6346e-01, -1.8080e+00],\n",
       "          [-3.4135e-01,  4.6377e-01, -4.9934e-01,  ..., -3.2617e-01,\n",
       "            3.4737e-01,  6.0806e-01],\n",
       "          [-9.6320e-01,  3.6988e-01, -1.1724e+00,  ...,  1.3694e-01,\n",
       "           -8.4960e-01, -7.4848e-01]],\n",
       "\n",
       "         [[ 2.7316e-01, -3.1403e-01,  1.0195e-01,  ...,  2.7107e-01,\n",
       "            2.4500e-01, -6.3062e-02],\n",
       "          [-4.0050e-01, -2.4306e+00,  3.4632e-03,  ...,  6.5701e-01,\n",
       "            1.6429e+00, -1.1010e+00],\n",
       "          [ 4.9272e-02, -6.3851e-01, -1.1208e+00,  ...,  9.9012e-01,\n",
       "            9.1692e-01, -1.8038e+00],\n",
       "          ...,\n",
       "          [-7.5512e-01, -1.3424e-01, -2.2085e-01,  ...,  6.3152e-01,\n",
       "            9.8529e-01,  6.1609e-01],\n",
       "          [-1.5096e-01,  8.6891e-01, -8.4786e-01,  ...,  4.4996e-01,\n",
       "            1.0761e+00,  7.3761e-01],\n",
       "          [-1.0859e+00, -1.9083e+00,  2.6600e-01,  ...,  1.1794e+00,\n",
       "            1.4066e+00, -5.7085e-01]]],\n",
       "\n",
       "\n",
       "        [[[-6.6505e-01,  5.0201e-02, -2.0597e-01,  ..., -2.1799e-01,\n",
       "            1.7665e-01, -8.1251e-02],\n",
       "          [-1.3709e+00, -1.2064e+00,  1.3008e-01,  ...,  4.1266e-01,\n",
       "           -1.1562e+00, -1.3032e-01],\n",
       "          [-9.3367e-01, -1.2449e+00, -1.4376e+00,  ...,  3.7338e-01,\n",
       "            9.6058e-01, -1.4504e-01],\n",
       "          ...,\n",
       "          [-9.0981e-01, -1.6660e+00, -9.0322e-01,  ..., -9.2820e-01,\n",
       "            9.8923e-01,  2.1988e-01],\n",
       "          [-5.1545e-01,  2.1257e-01, -5.9744e-01,  ..., -3.0145e-01,\n",
       "           -3.6068e-01, -1.5492e-01],\n",
       "          [ 1.4051e-01,  2.4129e-01, -1.3195e+00,  ..., -5.6572e-01,\n",
       "           -1.5320e-01, -8.8690e-01]],\n",
       "\n",
       "         [[ 3.7990e-01,  1.0530e+00,  1.3696e-01,  ...,  6.0863e-01,\n",
       "           -3.0163e-02, -9.4755e-02],\n",
       "          [ 1.1334e-02,  2.1083e+00,  3.4826e-01,  ...,  3.7859e-01,\n",
       "            6.1314e-01,  9.2821e-01],\n",
       "          [ 6.5634e-01,  4.6380e-01,  4.6287e-01,  ...,  1.6906e-01,\n",
       "           -5.2688e-01, -1.5186e+00],\n",
       "          ...,\n",
       "          [-2.7456e-01,  7.3189e-01,  7.0943e-01,  ...,  5.1593e-01,\n",
       "           -4.4623e-01, -2.8163e-01],\n",
       "          [ 3.5563e-01,  3.2304e-01,  1.9128e-01,  ..., -5.6214e-01,\n",
       "            2.9866e-01,  3.2873e-01],\n",
       "          [-2.1586e-01,  1.1336e+00,  4.1963e-01,  ...,  1.2350e-01,\n",
       "           -7.7675e-01, -5.9980e-01]],\n",
       "\n",
       "         [[ 8.4944e-02,  1.6510e-01, -1.6270e-01,  ...,  2.6079e-01,\n",
       "            3.9510e-01,  1.1643e+00],\n",
       "          [ 8.2744e-01, -8.0202e-01, -6.4630e-01,  ..., -3.3780e-01,\n",
       "           -4.1003e-01,  5.6065e-01],\n",
       "          [ 1.9824e+00,  8.0476e-01, -6.2764e-01,  ..., -4.4379e-01,\n",
       "           -6.7682e-02,  5.9786e-01],\n",
       "          ...,\n",
       "          [ 8.9229e-02,  6.5137e-01, -8.0774e-01,  ...,  2.1413e-01,\n",
       "           -1.0322e-01,  4.0203e-01],\n",
       "          [-1.8343e-01,  2.3967e-01, -6.9695e-01,  ..., -1.8553e-01,\n",
       "           -6.8361e-01,  1.1812e+00],\n",
       "          [ 6.8387e-02, -9.2944e-01, -6.4027e-01,  ...,  8.1074e-01,\n",
       "            1.7156e+00,  8.1915e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-3.2927e-01,  2.7549e-01, -1.2229e-01,  ...,  8.4761e-02,\n",
       "            1.4121e-01,  5.9524e-02],\n",
       "          [ 1.2524e+00,  6.8261e-02, -1.1619e+00,  ..., -4.6165e-01,\n",
       "           -1.2165e-01,  1.5591e+00],\n",
       "          [ 1.4843e+00, -4.9283e-01,  4.8654e-01,  ..., -3.5781e-01,\n",
       "           -1.9308e-02,  1.3569e+00],\n",
       "          ...,\n",
       "          [-7.9154e-01,  7.2566e-02, -1.2059e-01,  ...,  1.6013e+00,\n",
       "            2.0222e-02, -1.5826e-01],\n",
       "          [ 1.0961e+00, -5.0982e-01,  4.5396e-01,  ..., -2.1417e-01,\n",
       "            1.2020e-01,  4.7477e-01],\n",
       "          [ 1.0874e+00, -6.3139e-01,  6.5319e-01,  ..., -7.4378e-01,\n",
       "            3.3395e-01,  3.1536e-01]],\n",
       "\n",
       "         [[-1.4973e+00,  3.7447e-01, -7.6995e-02,  ...,  6.5077e-01,\n",
       "            5.0672e-01,  1.3044e+00],\n",
       "          [ 1.3323e-01, -1.2340e+00, -1.2805e+00,  ...,  3.7904e-01,\n",
       "            9.5483e-01,  3.7288e-01],\n",
       "          [-3.0217e-01, -9.3025e-01, -1.5310e+00,  ...,  1.8289e-01,\n",
       "           -3.8063e-01,  1.4495e+00],\n",
       "          ...,\n",
       "          [ 3.1213e-01, -4.4524e-01,  8.6051e-01,  ...,  3.0123e-01,\n",
       "           -9.4376e-01, -2.8777e-01],\n",
       "          [-2.9342e-01, -7.5385e-01, -4.3818e-02,  ..., -1.2165e+00,\n",
       "            7.0337e-01, -1.6117e+00],\n",
       "          [-9.9344e-01,  1.5081e+00, -3.1560e-02,  ...,  3.8717e-01,\n",
       "           -1.9249e+00, -8.7004e-01]],\n",
       "\n",
       "         [[ 2.6649e-01, -3.4917e-01,  8.1923e-02,  ...,  2.6750e-01,\n",
       "            3.0839e-01, -7.9346e-02],\n",
       "          [-1.3653e+00, -2.5276e+00,  2.2526e+00,  ...,  2.6397e+00,\n",
       "            1.4269e+00, -2.2167e-01],\n",
       "          [-5.4203e-01, -4.0894e-01, -4.2284e-01,  ...,  5.8294e-01,\n",
       "           -4.0605e-01,  1.3310e-01],\n",
       "          ...,\n",
       "          [ 1.6247e+00, -2.0465e-01, -1.3044e+00,  ...,  5.2421e-01,\n",
       "           -2.7665e-01, -8.5192e-02],\n",
       "          [ 1.6016e-01, -1.2473e+00, -6.5109e-02,  ...,  4.3295e-02,\n",
       "            1.2571e+00, -1.3745e+00],\n",
       "          [-1.0333e-01, -1.0075e+00,  1.7063e+00,  ...,  1.2205e-01,\n",
       "            1.3635e+00, -2.4362e-01]]],\n",
       "\n",
       "\n",
       "        [[[-6.5545e-01,  2.2324e-02, -2.3563e-01,  ..., -2.4275e-01,\n",
       "            1.8000e-01, -7.8724e-02],\n",
       "          [-6.7901e-01,  5.7810e-01, -6.5546e-01,  ...,  7.9150e-01,\n",
       "            6.9546e-01, -4.4282e-01],\n",
       "          [ 7.2048e-02,  1.6808e-01, -1.2536e+00,  ..., -2.0284e-01,\n",
       "            6.0289e-01,  3.5529e-01],\n",
       "          ...,\n",
       "          [-1.5219e+00, -8.2292e-01, -2.3042e-01,  ..., -1.1362e+00,\n",
       "            2.0255e+00,  8.0787e-02],\n",
       "          [ 1.5430e+00,  1.3432e-01, -8.9788e-01,  ...,  5.4653e-01,\n",
       "           -8.6746e-01, -4.2958e-01],\n",
       "          [-2.1552e+00, -4.3705e-01, -1.8439e+00,  ...,  3.2360e-01,\n",
       "            1.4084e+00,  3.4316e-01]],\n",
       "\n",
       "         [[ 3.7142e-01,  1.0553e+00,  1.2832e-01,  ...,  6.1187e-01,\n",
       "           -5.1856e-02, -1.0163e-01],\n",
       "          [ 4.7986e-01,  7.9887e-01, -9.3762e-02,  ...,  4.8713e-01,\n",
       "            3.6665e-01, -7.6067e-01],\n",
       "          [ 8.2490e-01, -6.8312e-01, -2.0828e-01,  ...,  1.2553e-02,\n",
       "           -1.8592e+00, -1.0218e+00],\n",
       "          ...,\n",
       "          [-9.7054e-02,  1.8336e+00,  1.3686e-01,  ...,  3.7153e-01,\n",
       "           -5.0175e-01, -8.1254e-01],\n",
       "          [ 7.1894e-01,  6.7605e-01,  1.5013e-02,  ..., -4.5631e-01,\n",
       "           -1.1943e+00, -8.0305e-01],\n",
       "          [ 6.3778e-01,  7.0502e-01,  5.2663e-01,  ..., -1.3573e-01,\n",
       "           -1.3096e+00, -2.4272e+00]],\n",
       "\n",
       "         [[ 5.5241e-02,  1.5979e-01, -1.5503e-01,  ...,  2.7393e-01,\n",
       "            4.0545e-01,  1.1600e+00],\n",
       "          [-2.4659e-03,  5.9305e-01, -5.4933e-01,  ...,  9.6980e-02,\n",
       "            2.3561e-01,  1.4091e+00],\n",
       "          [ 8.1823e-01,  8.0404e-01,  6.3352e-01,  ...,  8.8901e-02,\n",
       "            1.6688e-01,  1.3110e+00],\n",
       "          ...,\n",
       "          [ 6.2964e-01, -6.0033e-01, -5.0523e-01,  ...,  4.1813e-01,\n",
       "            7.7472e-01,  1.4371e+00],\n",
       "          [ 2.6790e-01, -9.1536e-01, -3.0966e-01,  ...,  4.3617e-01,\n",
       "            4.7323e-01,  1.0867e+00],\n",
       "          [ 8.9467e-02, -1.0537e+00, -1.1095e+00,  ...,  1.0819e+00,\n",
       "            1.0564e+00,  5.3892e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-3.0412e-01,  2.6283e-01, -1.4758e-01,  ...,  7.1489e-02,\n",
       "            1.4835e-01,  6.3154e-02],\n",
       "          [-4.4072e-01,  1.6359e+00, -4.1117e-01,  ...,  1.4818e+00,\n",
       "            6.5502e-01,  8.3334e-01],\n",
       "          [ 9.2691e-01,  1.8731e-01, -1.3368e+00,  ..., -2.4001e+00,\n",
       "            8.0771e-02,  2.0602e+00],\n",
       "          ...,\n",
       "          [ 1.2358e+00, -6.3574e-01, -6.5512e-01,  ..., -1.3798e+00,\n",
       "            2.7079e-01, -6.0532e-01],\n",
       "          [ 1.5669e-01, -2.4714e+00,  1.9365e-02,  ..., -2.2489e+00,\n",
       "           -1.4333e+00, -5.8078e-02],\n",
       "          [ 2.0242e+00, -1.4982e-02,  4.9099e-01,  ...,  4.7739e-01,\n",
       "           -6.5797e-01,  5.8022e-01]],\n",
       "\n",
       "         [[-1.5113e+00,  3.5705e-01, -8.4535e-02,  ...,  6.3212e-01,\n",
       "            5.2313e-01,  1.2954e+00],\n",
       "          [ 3.4843e-01, -2.9209e-01, -7.4211e-01,  ...,  1.1473e+00,\n",
       "           -1.6242e+00,  7.8340e-01],\n",
       "          [ 3.0061e-01,  1.1891e+00, -2.2079e+00,  ...,  2.1268e+00,\n",
       "           -1.4969e+00,  1.2887e+00],\n",
       "          ...,\n",
       "          [-5.4545e-01,  2.7155e-01, -4.8006e-02,  ..., -4.1918e-01,\n",
       "           -9.3510e-01,  4.6638e-01],\n",
       "          [-5.9257e-01, -2.3623e-01, -9.0067e-01,  ..., -1.7482e-01,\n",
       "            1.9578e-01,  2.1389e-01],\n",
       "          [-5.6411e-02,  2.7901e-01, -6.7464e-01,  ..., -4.8333e-01,\n",
       "           -1.4152e+00, -1.2496e+00]],\n",
       "\n",
       "         [[ 2.5635e-01, -3.4857e-01,  8.1171e-02,  ...,  2.7671e-01,\n",
       "            2.9050e-01, -6.9340e-02],\n",
       "          [ 9.4114e-01, -3.4930e-02, -2.4082e-01,  ..., -5.2099e-01,\n",
       "            7.1894e-01,  3.1050e+00],\n",
       "          [ 2.3487e+00, -1.1906e-01,  1.1345e+00,  ..., -3.2810e-01,\n",
       "            5.5668e-01,  1.0023e+00],\n",
       "          ...,\n",
       "          [ 1.1756e+00, -3.6660e-01, -6.8875e-02,  ..., -5.9614e-01,\n",
       "           -1.0932e+00,  5.2145e-01],\n",
       "          [ 5.9751e-01, -2.3802e+00, -1.1402e+00,  ...,  2.9009e-01,\n",
       "            1.1452e+00, -8.5016e-01],\n",
       "          [-9.7333e-01, -3.5510e-01,  3.2446e-02,  ...,  1.4728e-01,\n",
       "            8.9195e-01, -1.1143e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 1.4776e-01, -3.8590e-01, -2.6073e+00,  ..., -5.5130e-01,\n",
       "           -5.8441e-01, -6.9511e-01],\n",
       "          [ 3.4233e+00,  2.5308e-02, -3.4135e+00,  ..., -1.7325e+00,\n",
       "            6.3037e-01,  1.1040e+00],\n",
       "          [-2.9400e+00,  2.9799e-01, -4.2346e-01,  ...,  1.0707e+00,\n",
       "           -1.1548e+00,  2.8769e+00],\n",
       "          ...,\n",
       "          [ 1.2171e+00, -1.2523e+00,  4.9534e-01,  ..., -2.4558e+00,\n",
       "           -1.6548e+00, -3.4016e+00],\n",
       "          [-2.1376e+00, -2.0455e+00,  4.5565e+00,  ...,  3.1855e+00,\n",
       "           -1.6264e+00,  5.7071e+00],\n",
       "          [-4.4091e-01, -6.6018e-01, -5.7536e-01,  ..., -1.1394e+00,\n",
       "            3.3689e+00,  3.9815e+00]],\n",
       "\n",
       "         [[-1.8044e+00,  2.1724e+00, -1.7763e+00,  ...,  1.9361e-01,\n",
       "           -1.2330e+00,  3.5701e+00],\n",
       "          [-2.7568e+00,  5.4774e+00, -3.0292e+00,  ..., -2.0904e+00,\n",
       "            1.0375e+00,  3.8542e+00],\n",
       "          [-5.8188e-01,  8.1410e+00, -6.5779e-02,  ...,  2.7131e+00,\n",
       "           -1.5274e+00, -1.0431e+00],\n",
       "          ...,\n",
       "          [ 2.8078e+00,  1.8162e+00,  1.0342e+00,  ...,  2.8229e+00,\n",
       "            1.7181e+00,  3.3166e-01],\n",
       "          [ 3.1582e+00, -4.4279e-01, -4.5942e+00,  ...,  5.5977e-01,\n",
       "           -1.4053e+00,  1.0986e+00],\n",
       "          [ 4.1406e+00,  4.0671e-01,  2.0973e+00,  ..., -3.4526e+00,\n",
       "           -1.6503e-01, -5.8986e-01]],\n",
       "\n",
       "         [[-8.0889e-01,  1.9225e+00, -3.7391e+00,  ..., -3.9147e+00,\n",
       "           -1.0914e+00, -1.9307e+00],\n",
       "          [-3.9662e-01, -1.6777e+00, -3.3693e+00,  ..., -1.7250e+00,\n",
       "           -3.1363e+00,  3.4227e+00],\n",
       "          [ 1.6931e+00, -7.0586e-01,  8.6253e-01,  ..., -9.6575e-01,\n",
       "            4.5912e+00,  2.5124e+00],\n",
       "          ...,\n",
       "          [-4.8068e-02,  7.7006e-01, -5.4990e-02,  ...,  6.5813e-01,\n",
       "           -1.0296e+00,  1.7546e+00],\n",
       "          [ 1.1322e+00,  2.0522e-01,  2.8770e-01,  ...,  1.2025e+00,\n",
       "            4.7725e-01,  4.8867e-01],\n",
       "          [-4.2486e+00,  1.9652e+00, -2.7826e-01,  ..., -2.2064e+00,\n",
       "            2.9361e+00,  4.7446e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.7657e+00,  2.9197e+00, -4.4425e+00,  ..., -1.4629e+00,\n",
       "           -1.8276e+00, -9.9837e-01],\n",
       "          [ 1.0105e+00, -3.6081e+00, -2.1533e+00,  ..., -1.9643e+00,\n",
       "           -4.8026e-01, -1.1298e+00],\n",
       "          [ 1.1412e+00, -5.2390e-01, -2.4249e+00,  ..., -1.7870e-01,\n",
       "            3.8638e-02, -1.4962e+00],\n",
       "          ...,\n",
       "          [-1.0622e+00, -1.6476e+00,  3.5700e-01,  ..., -1.2453e+00,\n",
       "            2.1276e+00, -1.3192e+00],\n",
       "          [-1.0005e+00, -1.1380e+00,  8.8260e-01,  ..., -1.0445e+00,\n",
       "            1.4829e+00, -2.3527e+00],\n",
       "          [ 2.6064e+00,  9.5493e-01,  3.0247e+00,  ..., -3.2752e+00,\n",
       "            2.4958e+00, -6.4635e-01]],\n",
       "\n",
       "         [[ 1.5532e+00, -2.1424e-01, -7.2771e-02,  ...,  1.3022e+00,\n",
       "           -5.1210e-01, -1.0150e-01],\n",
       "          [-2.6404e+00,  4.0687e-01, -1.6843e+00,  ..., -2.9080e+00,\n",
       "           -3.0760e-01,  1.2654e-01],\n",
       "          [ 1.0769e-01, -2.4173e+00,  4.0773e-02,  ..., -1.7939e+00,\n",
       "           -4.1882e+00, -1.2626e-01],\n",
       "          ...,\n",
       "          [-1.3963e+00, -2.3914e+00, -1.8867e+00,  ...,  1.1648e+00,\n",
       "           -2.6822e+00, -3.4040e+00],\n",
       "          [-2.0672e+00,  1.1095e+00, -6.1649e-01,  ..., -3.3479e+00,\n",
       "           -8.9933e-01, -3.5373e+00],\n",
       "          [-1.2103e-01, -1.6089e+00,  1.0839e+00,  ...,  1.7476e+00,\n",
       "            1.3878e+00,  1.0778e+00]],\n",
       "\n",
       "         [[ 2.3463e+00, -8.9563e-01,  1.4256e+00,  ...,  3.4634e+00,\n",
       "            2.8074e+00,  2.4742e+00],\n",
       "          [ 5.2938e+00,  8.9569e-02,  1.1231e-01,  ...,  3.6554e+00,\n",
       "            3.4607e-01, -3.0231e-01],\n",
       "          [ 2.2528e+00, -6.1261e+00,  1.2105e-01,  ...,  8.4061e-01,\n",
       "            1.8033e+00,  2.5947e+00],\n",
       "          ...,\n",
       "          [ 7.3371e-01, -2.6840e+00,  1.7129e+00,  ..., -6.8827e-01,\n",
       "           -1.2287e+00,  1.8069e-01],\n",
       "          [ 7.3068e-01, -4.9586e+00, -6.3414e-01,  ...,  4.0789e-01,\n",
       "           -8.9105e-02,  2.2530e+00],\n",
       "          [-2.6770e-01,  2.8145e+00, -1.7058e+00,  ...,  2.8033e+00,\n",
       "            1.8138e+00,  2.6298e+00]]],\n",
       "\n",
       "\n",
       "        [[[-6.1446e-01,  6.0969e-01, -1.2261e+00,  ..., -7.5274e-01,\n",
       "           -4.1345e+00,  1.6584e+00],\n",
       "          [-4.1544e+00,  1.1116e+00,  1.4756e+00,  ...,  1.6797e+00,\n",
       "           -3.3787e+00,  2.0976e+00],\n",
       "          [ 5.8102e-01,  1.9156e-02,  2.3059e+00,  ...,  7.4180e-01,\n",
       "           -6.3411e-01,  3.7744e+00],\n",
       "          ...,\n",
       "          [ 2.1468e+00,  2.8657e-01, -6.5272e-01,  ..., -1.9628e+00,\n",
       "           -1.3118e+00, -1.4977e-01],\n",
       "          [-7.5472e-02, -3.1299e+00,  7.2173e-01,  ...,  1.5997e+00,\n",
       "            1.6845e-01,  9.1984e-01],\n",
       "          [-3.7012e-01, -8.1085e-01, -6.8294e-01,  ..., -1.2048e+00,\n",
       "            3.3967e+00,  3.8619e+00]],\n",
       "\n",
       "         [[ 4.2713e+00,  2.2338e+00,  1.9646e+00,  ..., -3.8337e-01,\n",
       "           -2.2837e-01, -1.1412e+00],\n",
       "          [ 4.6367e+00,  4.7284e+00,  2.3706e+00,  ...,  3.5459e+00,\n",
       "           -2.0858e+00, -1.7136e+00],\n",
       "          [ 3.3414e+00, -6.7512e-02, -2.8626e+00,  ...,  2.6330e-01,\n",
       "            1.7428e+00, -2.3274e+00],\n",
       "          ...,\n",
       "          [ 2.8867e+00,  2.2871e+00, -2.3569e+00,  ...,  3.0004e+00,\n",
       "            7.8129e-01, -2.0590e+00],\n",
       "          [ 2.3156e+00, -5.2270e-02, -2.7660e+00,  ..., -1.5522e+00,\n",
       "           -9.5079e-02,  1.3362e+00],\n",
       "          [ 4.1507e+00,  6.3162e-01,  2.0088e+00,  ..., -3.3916e+00,\n",
       "           -1.1288e-01, -5.0795e-01]],\n",
       "\n",
       "         [[-1.3814e+00,  3.8754e-01,  6.7554e-01,  ...,  1.1768e+00,\n",
       "            1.0500e+00, -1.6063e+00],\n",
       "          [ 4.8781e-01,  3.1004e-01,  2.7508e+00,  ..., -8.2583e-01,\n",
       "            3.5729e+00,  2.2651e+00],\n",
       "          [-9.8160e-01, -2.7829e+00, -5.7689e-02,  ...,  1.8675e+00,\n",
       "            3.4993e+00,  2.4215e+00],\n",
       "          ...,\n",
       "          [ 2.4131e-01, -3.0425e+00, -3.6123e+00,  ...,  2.3419e+00,\n",
       "            1.1646e+00,  8.9294e-01],\n",
       "          [ 7.1142e-01, -5.6543e-01, -4.8021e+00,  ...,  2.6067e+00,\n",
       "           -1.5545e+00,  3.3433e-01],\n",
       "          [-4.1731e+00,  1.9267e+00, -3.6486e-01,  ..., -2.1655e+00,\n",
       "            2.9412e+00,  3.4366e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.5346e-02,  1.4965e+00, -6.9476e-01,  ..., -1.3006e+00,\n",
       "            2.0432e+00,  1.6648e+00],\n",
       "          [ 7.1663e-01,  1.1008e+00, -9.7717e-01,  ...,  5.9440e-01,\n",
       "            1.1189e+00, -1.3300e+00],\n",
       "          [ 2.3985e-01,  4.5269e-01, -1.6356e+00,  ...,  3.1031e-01,\n",
       "            1.3290e+00,  2.7330e-01],\n",
       "          ...,\n",
       "          [ 8.2656e-01, -1.1223e+00, -4.1360e+00,  ..., -5.7670e+00,\n",
       "           -6.5595e-01,  1.9803e+00],\n",
       "          [-7.2569e-02, -2.5877e+00, -2.7312e-02,  ..., -3.8348e+00,\n",
       "            2.1725e-01, -1.5798e+00],\n",
       "          [ 2.7277e+00,  9.1027e-01,  2.9940e+00,  ..., -3.3573e+00,\n",
       "            2.5147e+00, -5.6020e-01]],\n",
       "\n",
       "         [[-2.0422e+00, -6.2646e-01,  4.6028e-02,  ..., -1.1337e+00,\n",
       "            1.0630e+00, -2.0392e+00],\n",
       "          [-3.2332e+00, -1.7807e+00,  1.4520e+00,  ..., -4.6529e-01,\n",
       "           -5.7314e+00, -4.5900e-01],\n",
       "          [-1.3333e+00,  6.3203e-01,  1.3019e+00,  ..., -1.5885e+00,\n",
       "           -1.4978e+00, -1.1662e+00],\n",
       "          ...,\n",
       "          [ 7.4771e-01,  2.4177e+00, -5.1187e+00,  ..., -1.2494e+00,\n",
       "           -4.5699e-01,  8.5718e-01],\n",
       "          [ 5.5932e-01,  1.0453e-01, -1.9681e+00,  ...,  1.4599e+00,\n",
       "            4.5331e-01,  1.3417e+00],\n",
       "          [-9.9863e-02, -1.7524e+00,  1.0220e+00,  ...,  1.7958e+00,\n",
       "            1.4253e+00,  1.1817e+00]],\n",
       "\n",
       "         [[ 3.2779e+00,  2.6040e-01,  7.0471e-01,  ..., -5.9464e-01,\n",
       "            4.1724e+00,  1.1527e+00],\n",
       "          [ 3.8386e-01, -8.0022e+00,  2.4420e+00,  ..., -2.1382e+00,\n",
       "            2.5963e+00,  3.1238e-01],\n",
       "          [ 1.3839e+00, -5.4834e+00, -1.0138e+00,  ..., -2.1017e+00,\n",
       "            8.6097e-01, -3.8874e-01],\n",
       "          ...,\n",
       "          [ 3.1485e+00,  2.1425e+00,  2.6416e+00,  ..., -4.5935e-02,\n",
       "            2.3362e-01,  1.7440e+00],\n",
       "          [ 3.2040e+00,  2.8349e+00,  3.2105e+00,  ...,  1.4717e+00,\n",
       "           -2.4163e+00,  1.6016e+00],\n",
       "          [-1.7510e-01,  2.7818e+00, -1.7568e+00,  ...,  2.9360e+00,\n",
       "            1.7295e+00,  2.6651e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 2.1395e+00, -8.6014e-01, -1.3660e+00,  ..., -2.4185e-01,\n",
       "           -2.1949e+00,  3.8292e-01],\n",
       "          [-5.9146e-01, -3.6521e+00,  3.5104e+00,  ...,  3.2594e+00,\n",
       "           -1.3144e+00,  3.3233e+00],\n",
       "          [-1.1737e+00, -2.0109e-01,  3.2912e+00,  ...,  3.3290e+00,\n",
       "           -2.9837e+00,  2.1342e+00],\n",
       "          ...,\n",
       "          [-3.9832e+00, -4.7633e+00,  5.5902e+00,  ...,  3.0005e+00,\n",
       "            8.2806e-01,  3.6095e+00],\n",
       "          [-1.6718e+00, -3.7654e+00,  2.2000e+00,  ...,  3.7801e+00,\n",
       "            6.6409e-01, -7.3709e-01],\n",
       "          [-4.2665e-01, -7.7364e-01, -5.7822e-01,  ..., -1.1278e+00,\n",
       "            3.3901e+00,  3.9099e+00]],\n",
       "\n",
       "         [[ 3.4845e+00,  2.5853e+00, -2.5516e+00,  ..., -6.6838e-01,\n",
       "           -2.7335e-01, -1.3900e+00],\n",
       "          [ 1.2406e+00,  2.6509e+00, -1.0825e+00,  ...,  1.8631e+00,\n",
       "           -3.2786e-01, -7.2361e-01],\n",
       "          [-3.0701e+00,  8.1253e-01, -2.6951e+00,  ...,  7.5912e-01,\n",
       "           -3.6519e+00, -2.1670e-01],\n",
       "          ...,\n",
       "          [ 3.5031e+00, -1.2924e+00, -1.3746e+00,  ...,  2.9515e-01,\n",
       "            1.2905e+00, -6.8960e-01],\n",
       "          [ 4.2015e+00, -1.1180e+00, -1.6564e-01,  ...,  6.3390e-01,\n",
       "            2.0941e+00,  1.5972e+00],\n",
       "          [ 4.1586e+00,  3.9437e-01,  2.1581e+00,  ..., -3.4337e+00,\n",
       "           -1.2499e-01, -6.0422e-01]],\n",
       "\n",
       "         [[-1.6003e+00,  1.5870e+00, -5.4139e-01,  ..., -2.4838e-01,\n",
       "            2.2172e+00, -1.6598e+00],\n",
       "          [ 7.4929e-01,  2.0046e+00, -2.1887e+00,  ...,  2.3001e+00,\n",
       "            4.9808e+00, -2.6905e+00],\n",
       "          [-5.1718e-02,  3.8938e+00,  1.6502e+00,  ..., -8.9297e-01,\n",
       "           -2.2258e-01, -3.1692e+00],\n",
       "          ...,\n",
       "          [-1.1070e+00,  3.0943e+00,  4.3103e+00,  ...,  8.1314e-01,\n",
       "            3.2270e-01, -2.6613e+00],\n",
       "          [ 9.6494e-01,  5.5263e-03, -2.6558e-02,  ...,  4.9442e+00,\n",
       "           -1.3565e+00,  4.4036e-01],\n",
       "          [-4.1761e+00,  1.9612e+00, -3.6855e-01,  ..., -2.1672e+00,\n",
       "            2.9261e+00, -7.8836e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.0150e-01, -2.0830e+00, -8.9012e-01,  ...,  1.6600e+00,\n",
       "           -3.2700e-01, -6.0548e-02],\n",
       "          [ 1.2135e+00, -1.1178e+00, -2.4507e+00,  ...,  1.4551e+00,\n",
       "           -3.8253e-01, -1.5397e+00],\n",
       "          [-1.2115e+00, -2.7978e+00, -2.1086e+00,  ...,  2.7299e-01,\n",
       "            1.4446e+00,  9.4550e-01],\n",
       "          ...,\n",
       "          [-3.1112e-01,  3.3774e-01, -2.5022e+00,  ..., -1.2061e+00,\n",
       "            1.1958e+00, -2.1387e+00],\n",
       "          [-9.9032e-02, -7.6362e-01,  9.0328e-01,  ..., -2.5728e-01,\n",
       "            1.5727e+00, -2.0488e+00],\n",
       "          [ 2.7265e+00,  9.4711e-01,  2.9968e+00,  ..., -3.1917e+00,\n",
       "            2.5165e+00, -5.1136e-01]],\n",
       "\n",
       "         [[-5.5581e-01,  9.3019e-02, -5.9973e-01,  ..., -3.3442e-01,\n",
       "            5.7101e-01,  1.5363e+00],\n",
       "          [-9.4111e-01,  2.9335e+00,  3.2071e+00,  ...,  2.2317e-01,\n",
       "           -1.5045e+00, -3.9544e+00],\n",
       "          [-7.6342e+00, -2.5514e+00,  3.1164e+00,  ...,  1.9627e-01,\n",
       "           -3.8359e+00,  7.6993e+00],\n",
       "          ...,\n",
       "          [-3.9044e+00, -8.5561e-01,  1.1500e+00,  ...,  5.3881e-01,\n",
       "           -2.7461e+00,  5.8222e-01],\n",
       "          [-3.0935e+00,  2.3154e+00, -4.4613e-01,  ..., -2.9252e+00,\n",
       "            2.6788e+00, -1.7150e+00],\n",
       "          [-1.6993e-01, -1.5756e+00,  1.1217e+00,  ...,  1.6747e+00,\n",
       "            1.5193e+00,  1.1406e+00]],\n",
       "\n",
       "         [[-2.5943e-01,  1.1067e+00, -2.5895e+00,  ..., -3.3235e-01,\n",
       "            3.4531e+00,  1.6492e+00],\n",
       "          [-4.2237e+00, -2.4370e+00, -3.1147e+00,  ...,  4.6760e-01,\n",
       "            3.4706e-01, -3.0645e-01],\n",
       "          [-3.7994e+00, -1.1702e+00, -6.4999e-01,  ..., -3.1318e+00,\n",
       "            7.1759e-01, -5.2182e-02],\n",
       "          ...,\n",
       "          [-9.1218e-01,  8.5863e-01, -8.5248e-01,  ...,  9.8216e-01,\n",
       "           -8.1172e-01,  4.3820e-01],\n",
       "          [-1.6033e+00, -2.1395e+00, -1.6795e+00,  ..., -6.0236e-01,\n",
       "            1.1879e+00,  5.5942e-01],\n",
       "          [-2.3593e-01,  2.8345e+00, -1.7562e+00,  ...,  2.8264e+00,\n",
       "            1.7766e+00,  2.6762e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-5.9442e-01,  7.6636e-01, -1.3105e+00,  ...,  3.1075e-01,\n",
       "            1.3579e+00, -3.8265e-02],\n",
       "          [-1.8454e-01, -1.7614e+00, -1.0790e+00,  ...,  8.8036e-02,\n",
       "           -2.3126e+00, -1.0807e+00],\n",
       "          [-2.7726e+00,  3.6382e-01,  7.3917e-02,  ...,  5.1345e-01,\n",
       "            1.0777e+00, -5.2629e-01],\n",
       "          ...,\n",
       "          [-7.7116e-01, -2.3945e+00,  5.3774e-01,  ..., -1.2976e+00,\n",
       "           -1.0292e+00,  2.0603e+00],\n",
       "          [ 6.0894e-01,  3.0307e-01,  1.5282e+00,  ...,  1.0652e+00,\n",
       "           -8.9821e-02,  1.1455e+00],\n",
       "          [ 2.1987e-01, -1.1355e-01,  1.9137e-01,  ..., -3.8214e-01,\n",
       "           -1.1076e-01,  1.9656e-02]],\n",
       "\n",
       "         [[ 3.4908e-01, -1.4671e+00,  5.1508e-01,  ...,  2.5933e+00,\n",
       "            1.3374e+00,  2.2064e+00],\n",
       "          [ 2.5522e+00,  3.0600e-01, -9.8823e-01,  ..., -8.6651e-01,\n",
       "           -1.5926e+00,  1.8406e+00],\n",
       "          [-1.9230e+00, -1.7502e+00, -4.9920e-01,  ..., -5.9111e-01,\n",
       "           -3.6002e+00,  9.1208e-01],\n",
       "          ...,\n",
       "          [ 4.9968e-01,  6.4585e-01, -6.9581e-01,  ...,  2.6830e-01,\n",
       "            7.0620e-01, -3.4323e-01],\n",
       "          [ 8.4228e-01, -3.1945e-01,  1.0899e+00,  ..., -2.7128e-01,\n",
       "            1.4588e-01, -5.4415e-01],\n",
       "          [ 5.4567e-01, -4.9064e-01,  3.9261e-01,  ..., -1.5403e-01,\n",
       "           -2.4472e-01,  8.5593e-01]],\n",
       "\n",
       "         [[ 2.0723e+00,  1.7413e-01, -6.5932e-01,  ..., -1.6081e-01,\n",
       "           -1.9320e+00, -1.2201e+00],\n",
       "          [ 1.4621e+00,  1.9670e+00,  2.2505e+00,  ..., -4.9995e-01,\n",
       "            1.3549e+00,  1.4509e+00],\n",
       "          [ 2.9948e-01,  3.3342e-02, -9.9169e-01,  ..., -6.7262e-01,\n",
       "           -6.1044e-01,  1.3106e+00],\n",
       "          ...,\n",
       "          [ 4.5226e-01,  1.1326e+00,  1.5135e-02,  ..., -1.5276e+00,\n",
       "            1.8240e+00,  1.6923e+00],\n",
       "          [ 4.1524e-01, -7.8498e-02, -3.5234e-01,  ..., -2.9228e-01,\n",
       "            3.1541e-01,  3.2464e-02],\n",
       "          [ 1.5664e+00,  3.5889e-01, -1.7780e-01,  ...,  2.6740e-01,\n",
       "           -4.7250e-01,  3.1498e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.2624e-02, -1.6314e+00, -1.4762e+00,  ...,  9.4136e-01,\n",
       "            6.5033e-02,  1.4994e+00],\n",
       "          [ 2.0668e+00, -1.3033e-01,  9.9118e-01,  ..., -3.4082e-01,\n",
       "            1.7084e-01,  2.2581e+00],\n",
       "          [ 2.8211e+00, -2.6597e+00,  4.9495e-01,  ...,  8.4331e-01,\n",
       "           -1.3317e+00,  7.8789e-01],\n",
       "          ...,\n",
       "          [-5.5944e-01, -3.6804e-01,  1.0905e+00,  ...,  1.1649e+00,\n",
       "            6.6614e-01, -7.4591e-01],\n",
       "          [ 1.8617e-01, -7.9267e-01,  2.5939e+00,  ...,  3.0308e-01,\n",
       "            7.0392e-01,  1.8135e+00],\n",
       "          [ 3.4720e-02,  2.9697e-01,  1.3220e+00,  ...,  5.7469e-01,\n",
       "           -1.3088e+00, -1.0786e+00]],\n",
       "\n",
       "         [[ 2.1035e+00,  1.5402e+00,  8.4233e-01,  ..., -8.3585e-01,\n",
       "            6.0509e-01, -3.3842e+00],\n",
       "          [-3.6432e-02,  1.2876e+00, -1.7613e+00,  ...,  5.5078e-01,\n",
       "           -2.2692e+00,  3.9888e-01],\n",
       "          [ 1.3319e+00,  8.6798e-01, -8.7515e-01,  ..., -1.8499e+00,\n",
       "            1.2284e+00,  3.3102e-01],\n",
       "          ...,\n",
       "          [ 4.3084e-01,  3.3967e+00, -1.2248e+00,  ..., -1.4894e-01,\n",
       "           -6.5893e-01, -7.7556e-01],\n",
       "          [ 1.7444e+00,  1.0124e+00,  6.4334e-01,  ...,  4.9610e-02,\n",
       "            9.2603e-01,  1.3741e+00],\n",
       "          [-3.6792e-01, -2.8444e-01, -4.4342e-01,  ...,  4.5832e-01,\n",
       "           -2.4794e-01, -4.1583e-01]],\n",
       "\n",
       "         [[ 1.3237e+00,  5.0660e-01, -1.6381e+00,  ..., -1.1542e-01,\n",
       "           -1.9974e+00, -1.1228e+00],\n",
       "          [ 6.5624e-01, -5.1457e-01, -3.2005e-01,  ...,  4.1453e-01,\n",
       "           -8.5539e-01,  2.3673e-01],\n",
       "          [-9.4833e-01,  5.3782e-01,  4.3230e-01,  ...,  1.0063e+00,\n",
       "           -2.0078e-01, -3.5259e-01],\n",
       "          ...,\n",
       "          [ 6.9451e-01,  6.5161e-01,  7.5884e-01,  ..., -5.5520e-01,\n",
       "           -1.9003e+00,  5.3150e-01],\n",
       "          [ 4.0395e-01,  3.4902e-01, -3.4321e-01,  ..., -7.5814e-01,\n",
       "           -3.0246e-01, -3.0263e-01],\n",
       "          [-7.5898e-02, -2.0084e-01,  3.4787e-01,  ...,  5.2993e-01,\n",
       "           -1.1966e-01,  7.0114e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.3601e-01, -1.6636e+00, -6.2541e-01,  ...,  4.7788e-01,\n",
       "           -1.8446e+00, -6.2495e-04],\n",
       "          [-2.3924e+00,  8.1517e-01,  1.9092e-01,  ...,  9.1402e-01,\n",
       "            1.7878e+00,  1.0606e+00],\n",
       "          [-1.0027e+00,  6.0439e-01,  1.1742e+00,  ..., -3.3328e-01,\n",
       "           -2.3268e-02, -1.7201e-01],\n",
       "          ...,\n",
       "          [-5.2473e-01,  4.2416e-01, -7.1634e-01,  ..., -1.4772e+00,\n",
       "           -2.4028e-01,  5.3703e-01],\n",
       "          [ 1.1058e+00, -9.9048e-01,  1.3562e+00,  ...,  1.4005e-01,\n",
       "           -2.5096e-01, -1.0442e-01],\n",
       "          [ 2.3588e-01, -7.6224e-02,  2.0441e-01,  ..., -3.8334e-01,\n",
       "           -1.3056e-01,  6.2605e-02]],\n",
       "\n",
       "         [[-2.1930e-01, -1.4458e+00,  6.6494e-01,  ...,  5.4858e-01,\n",
       "           -1.1818e+00,  1.4850e+00],\n",
       "          [-2.3174e+00, -2.7376e+00, -8.3343e-01,  ..., -6.7275e-01,\n",
       "           -3.4989e+00,  1.3225e+00],\n",
       "          [-2.2267e-01, -9.2280e-01, -1.9963e+00,  ..., -1.9243e+00,\n",
       "           -3.3221e+00,  1.3869e+00],\n",
       "          ...,\n",
       "          [-2.6605e+00, -1.2184e+00,  1.5664e+00,  ...,  8.2127e-01,\n",
       "           -2.9875e+00,  2.4370e-01],\n",
       "          [-4.4291e-01,  2.6133e-01,  3.8321e-01,  ...,  4.7054e-01,\n",
       "           -6.7934e-01,  1.0904e+00],\n",
       "          [ 4.7472e-01, -4.7127e-01,  3.2582e-01,  ..., -6.7431e-02,\n",
       "           -3.2521e-01,  9.3620e-01]],\n",
       "\n",
       "         [[-5.6356e-01,  3.5155e+00,  3.2969e+00,  ..., -4.2180e-01,\n",
       "            7.2385e-01,  1.0609e+00],\n",
       "          [ 3.4491e-01, -1.8468e+00, -5.8007e-01,  ..., -1.4913e+00,\n",
       "            2.9806e-01,  1.4324e+00],\n",
       "          [-3.1064e-01, -8.2859e-02,  1.1206e+00,  ..., -1.4213e+00,\n",
       "           -9.7925e-02,  6.0025e-01],\n",
       "          ...,\n",
       "          [-3.6917e-01, -9.9924e-01, -6.8609e-01,  ..., -3.4429e-01,\n",
       "           -7.1598e-02, -1.9714e-02],\n",
       "          [-8.0719e-01,  5.5437e-01,  1.7131e-01,  ...,  1.4002e-01,\n",
       "            6.5688e-01,  1.8495e+00],\n",
       "          [ 1.5521e+00,  3.9824e-01, -1.5391e-01,  ...,  2.7321e-01,\n",
       "           -4.3959e-01,  3.3620e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.6634e+00,  1.2189e+00,  1.2900e+00,  ...,  6.1943e-01,\n",
       "           -5.1883e-01,  2.2623e+00],\n",
       "          [ 1.7136e+00, -1.2519e+00,  9.1535e-01,  ...,  5.3000e-01,\n",
       "           -1.4535e+00, -4.6372e-01],\n",
       "          [ 8.8713e-01, -1.1393e+00,  2.4067e+00,  ...,  1.9947e+00,\n",
       "           -1.1508e+00,  2.4367e+00],\n",
       "          ...,\n",
       "          [ 1.2005e+00, -1.6353e-01,  1.6088e-01,  ...,  1.2956e+00,\n",
       "           -1.5954e+00,  3.4231e+00],\n",
       "          [ 7.0136e-01,  1.2841e-01,  4.6890e-01,  ...,  7.5717e-01,\n",
       "            2.3087e-01,  1.1238e+00],\n",
       "          [ 6.9437e-02,  2.6025e-01,  1.3074e+00,  ...,  6.3077e-01,\n",
       "           -1.3252e+00, -1.1087e+00]],\n",
       "\n",
       "         [[ 2.2807e+00,  1.7984e-01, -1.5226e+00,  ...,  4.3036e+00,\n",
       "           -1.3602e+00, -2.1756e+00],\n",
       "          [ 1.8200e+00,  1.4292e+00, -9.7929e-01,  ..., -1.2621e+00,\n",
       "            2.4526e+00, -9.5254e-01],\n",
       "          [ 1.2773e+00,  1.6036e-01,  7.4747e-01,  ..., -2.0332e+00,\n",
       "            2.9917e+00, -2.1563e+00],\n",
       "          ...,\n",
       "          [-2.3469e-01,  3.4393e-01,  1.1000e+00,  ...,  3.2223e+00,\n",
       "            1.5659e+00,  2.7794e+00],\n",
       "          [-2.0317e-01, -2.0126e-02,  2.5818e-01,  ...,  1.3796e-01,\n",
       "           -4.8130e-01,  7.1430e-01],\n",
       "          [-4.3252e-01, -3.1199e-01, -4.5113e-01,  ...,  5.0245e-01,\n",
       "           -2.6069e-01, -3.9418e-01]],\n",
       "\n",
       "         [[ 1.0800e+00,  1.2176e+00, -2.3510e+00,  ...,  1.1079e+00,\n",
       "            1.0729e+00, -6.6198e-01],\n",
       "          [-1.0445e+00,  1.6850e+00, -1.4043e-01,  ...,  9.0406e-01,\n",
       "           -7.5884e-01,  6.0388e-01],\n",
       "          [ 2.3929e-01, -4.7362e-01,  3.1190e-02,  ...,  4.9440e-01,\n",
       "           -3.2453e-01, -1.3671e+00],\n",
       "          ...,\n",
       "          [-6.8643e-01,  4.8943e-01, -2.0424e-01,  ..., -7.0748e-01,\n",
       "           -5.0707e-01, -1.7984e+00],\n",
       "          [-4.2225e-01, -3.2629e-01, -3.3885e-01,  ...,  3.7270e-02,\n",
       "            1.1860e+00, -6.7604e-01],\n",
       "          [-1.1161e-01, -2.1333e-01,  3.2441e-01,  ...,  5.5051e-01,\n",
       "           -1.0724e-01,  6.4339e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.5248e-01, -2.8390e+00, -9.0150e-01,  ...,  1.0698e+00,\n",
       "           -4.4247e-01, -1.4072e+00],\n",
       "          [-1.4733e+00,  2.9425e-01,  6.4419e-01,  ...,  5.2905e-01,\n",
       "           -1.8348e+00, -2.1644e-01],\n",
       "          [-1.0063e+00, -3.7012e-03,  2.2179e-01,  ...,  2.8168e+00,\n",
       "           -1.7526e-01, -2.8510e+00],\n",
       "          ...,\n",
       "          [-2.8090e+00,  1.8628e+00,  3.2870e-01,  ...,  3.3771e+00,\n",
       "            1.4429e+00,  1.3811e+00],\n",
       "          [ 1.3980e+00, -3.2865e-01,  4.7388e-01,  ..., -7.5722e-01,\n",
       "           -8.9651e-01,  2.6069e-01],\n",
       "          [ 2.2487e-01, -8.5772e-02,  1.9301e-01,  ..., -4.3582e-01,\n",
       "           -1.7715e-01, -6.8998e-03]],\n",
       "\n",
       "         [[ 3.2398e-01, -1.5139e+00, -1.6149e+00,  ..., -3.7803e-01,\n",
       "            2.1763e+00,  2.2567e+00],\n",
       "          [ 2.1321e-01,  1.7343e+00,  4.5992e-01,  ...,  7.0602e-01,\n",
       "           -1.3466e+00,  9.3084e-01],\n",
       "          [ 5.7610e-01, -2.1132e+00,  3.3653e+00,  ...,  2.5661e-01,\n",
       "           -5.7552e+00, -1.3508e+00],\n",
       "          ...,\n",
       "          [-3.1082e+00, -4.1454e+00, -2.0466e-01,  ..., -6.2324e-02,\n",
       "           -1.9170e+00, -1.0347e+00],\n",
       "          [-4.3396e-01,  3.6004e-01,  5.3390e-01,  ..., -3.4094e-01,\n",
       "           -7.5013e-01, -3.5139e-01],\n",
       "          [ 4.8148e-01, -4.5960e-01,  3.4351e-01,  ..., -1.9831e-01,\n",
       "           -2.9225e-01,  8.7656e-01]],\n",
       "\n",
       "         [[-3.6579e-01,  9.1646e-01,  1.6830e+00,  ..., -3.5533e-01,\n",
       "           -1.0018e+00,  2.4565e-01],\n",
       "          [ 2.7980e-01,  6.5620e-02, -9.4123e-02,  ...,  1.6393e+00,\n",
       "           -4.1668e-01,  5.5077e-01],\n",
       "          [ 2.4260e+00, -7.8745e-01, -1.6714e+00,  ...,  1.7455e+00,\n",
       "            9.6072e-02,  1.1554e+00],\n",
       "          ...,\n",
       "          [ 1.7164e+00, -2.4342e+00, -1.4034e-01,  ..., -1.0142e+00,\n",
       "           -1.9489e-01,  1.3371e+00],\n",
       "          [ 1.2263e+00,  3.6286e-01, -1.0076e+00,  ..., -3.8301e-01,\n",
       "            6.6496e-01, -7.2966e-03],\n",
       "          [ 1.6213e+00,  3.6730e-01, -1.8018e-01,  ...,  2.9166e-01,\n",
       "           -4.2898e-01,  3.0139e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.2988e-01,  8.7425e-01,  3.5175e-01,  ..., -1.0012e-01,\n",
       "            1.5545e+00,  9.6035e-02],\n",
       "          [ 2.4776e-01, -4.1722e-02,  1.0887e+00,  ...,  3.7343e-01,\n",
       "           -1.0585e+00,  1.3940e+00],\n",
       "          [-7.2302e-01, -4.5824e-01,  1.1394e+00,  ..., -3.6682e-01,\n",
       "           -1.7068e+00, -1.3629e-01],\n",
       "          ...,\n",
       "          [-1.7622e+00, -2.7388e+00,  3.1464e+00,  ..., -2.6179e-01,\n",
       "            1.0104e+00,  8.8339e-01],\n",
       "          [ 3.2033e-01, -7.7966e-02,  1.2768e+00,  ..., -2.6858e-01,\n",
       "            1.7368e-01,  1.3899e+00],\n",
       "          [-1.5068e-02,  2.5379e-01,  1.2628e+00,  ...,  5.8868e-01,\n",
       "           -1.3326e+00, -1.0716e+00]],\n",
       "\n",
       "         [[ 5.2003e-01,  1.6607e+00, -2.4927e-01,  ...,  8.4543e-04,\n",
       "           -3.3950e-01,  1.3682e+00],\n",
       "          [ 1.9423e-01,  8.2227e-01, -1.3491e-01,  ..., -1.6687e+00,\n",
       "           -6.0477e-01, -2.0534e+00],\n",
       "          [ 1.3222e+00,  1.5320e+00,  1.3503e-01,  ...,  8.2329e-01,\n",
       "           -6.4262e-01, -7.1756e-01],\n",
       "          ...,\n",
       "          [-7.3711e-01, -2.0153e+00, -2.1359e+00,  ..., -4.3777e-01,\n",
       "            9.0008e-01, -5.5982e-01],\n",
       "          [ 1.1381e+00, -6.7635e-01,  1.5281e+00,  ..., -5.1050e-02,\n",
       "            1.4548e+00, -7.2442e-01],\n",
       "          [-3.5341e-01, -3.3851e-01, -4.2309e-01,  ...,  5.0975e-01,\n",
       "           -2.7420e-01, -4.3511e-01]],\n",
       "\n",
       "         [[ 1.2037e+00,  1.3948e-01,  1.8048e+00,  ..., -1.0050e+00,\n",
       "            7.9588e-02, -1.7123e-01],\n",
       "          [ 1.6098e+00,  9.8734e-01,  1.2959e+00,  ..., -1.1833e+00,\n",
       "           -1.1167e+00,  5.2846e-01],\n",
       "          [ 5.5901e-01,  3.2381e+00,  5.2329e-01,  ..., -7.0359e-01,\n",
       "            1.0211e+00,  5.7632e-01],\n",
       "          ...,\n",
       "          [-5.5253e-01, -6.6382e-02,  6.5959e-01,  ..., -1.8857e-01,\n",
       "            6.5046e-01, -2.0669e+00],\n",
       "          [ 3.0012e-01, -3.4057e-01,  1.5793e-01,  ..., -7.1723e-01,\n",
       "            7.5509e-02, -4.2296e-01],\n",
       "          [-9.5561e-02, -2.2351e-01,  2.9537e-01,  ...,  5.6145e-01,\n",
       "           -1.2365e-01,  6.8528e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-1.3465e+00, -1.7829e+00,  2.0509e-01,  ..., -9.8697e-01,\n",
       "           -5.5894e-01, -1.1137e+00],\n",
       "          [ 5.6415e-01, -6.7654e-01, -2.2225e-01,  ...,  3.0786e-01,\n",
       "            5.9939e-01, -3.6994e-01],\n",
       "          [ 1.2067e-02, -6.4606e-01, -3.3904e-01,  ...,  4.6279e-01,\n",
       "            9.4971e-01, -1.3573e+00],\n",
       "          ...,\n",
       "          [ 1.1844e+00, -1.2324e+00, -8.2178e-01,  ...,  1.6737e+00,\n",
       "           -4.0991e-01, -1.4518e+00],\n",
       "          [ 1.1963e+00, -1.0313e+00, -9.3486e-02,  ..., -6.2787e-01,\n",
       "           -9.9252e-01, -1.7811e+00],\n",
       "          [ 1.4129e+00, -1.5447e-01, -7.1587e-01,  ..., -3.7797e-01,\n",
       "           -7.3684e-01, -2.2740e+00]],\n",
       "\n",
       "         [[ 1.6236e+00, -4.4947e-01,  1.8610e+00,  ...,  2.3850e-01,\n",
       "            4.8569e-01, -9.0377e-01],\n",
       "          [ 4.6505e-01, -5.8693e-02,  1.7935e+00,  ..., -7.0859e-01,\n",
       "            7.4887e-01, -8.4511e-01],\n",
       "          [-1.3376e-01, -4.1664e-02,  1.6274e+00,  ..., -4.8422e-01,\n",
       "            4.7511e-01, -9.9393e-01],\n",
       "          ...,\n",
       "          [ 1.5858e-01,  5.8026e-02,  8.7756e-01,  ..., -1.6815e+00,\n",
       "           -3.6455e-01, -5.1239e-01],\n",
       "          [ 1.4626e+00, -6.2122e-01,  7.9911e-02,  ...,  4.4857e-01,\n",
       "            1.3402e+00, -2.7384e-01],\n",
       "          [ 1.3052e+00,  1.6566e-01,  1.3616e-01,  ..., -8.9261e-02,\n",
       "           -4.5254e-02,  5.4956e-01]],\n",
       "\n",
       "         [[ 2.2243e-01, -7.6738e-01,  4.7152e-01,  ..., -6.9537e-01,\n",
       "            1.6363e+00,  5.2678e-02],\n",
       "          [-1.0993e+00, -3.3466e-01,  9.8926e-02,  ..., -1.3192e-01,\n",
       "           -8.4147e-01,  1.1772e-01],\n",
       "          [-1.0870e+00, -1.1900e+00,  1.1070e+00,  ...,  1.8156e-01,\n",
       "           -5.4384e-01, -7.7336e-01],\n",
       "          ...,\n",
       "          [-1.0598e+00,  7.7522e-01,  5.9787e-01,  ..., -1.4257e-01,\n",
       "            6.7722e-01, -8.4039e-01],\n",
       "          [-1.4791e+00,  9.9223e-01, -8.9566e-01,  ...,  9.3394e-01,\n",
       "           -1.5604e-01, -3.0277e-01],\n",
       "          [-5.1949e-02, -1.3298e-01, -4.2548e-01,  ...,  2.9724e-01,\n",
       "           -1.0388e+00,  6.1173e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.4289e+00, -1.0303e+00, -6.9659e-01,  ...,  4.8257e-01,\n",
       "            2.6117e-01, -7.8986e-01],\n",
       "          [-4.4836e-01, -1.7408e-02,  4.7254e-01,  ...,  1.3720e+00,\n",
       "           -9.9680e-01, -5.5070e-01],\n",
       "          [-2.1468e-01, -9.3435e-01,  6.2040e-01,  ...,  5.4743e-01,\n",
       "           -1.8745e+00, -2.0104e-01],\n",
       "          ...,\n",
       "          [-5.3481e-01, -8.5520e-01,  1.5789e+00,  ...,  1.0547e+00,\n",
       "            1.6317e+00,  1.0795e+00],\n",
       "          [ 2.8126e-01,  3.3272e-01,  1.5304e+00,  ...,  1.0072e+00,\n",
       "            7.9421e-01,  1.1969e+00],\n",
       "          [ 4.8733e-01, -1.4566e+00,  9.4535e-01,  ...,  9.2519e-01,\n",
       "           -7.3049e-01,  2.8673e+00]],\n",
       "\n",
       "         [[ 4.7602e-01,  1.4065e+00,  1.6059e-01,  ..., -4.8658e-01,\n",
       "            1.2046e+00,  5.3017e-01],\n",
       "          [ 1.8118e-01,  2.5875e-01, -4.3827e-01,  ...,  2.1733e-01,\n",
       "            7.5847e-01,  3.4850e-01],\n",
       "          [-3.6902e-02,  1.4952e-01, -1.3473e-01,  ...,  8.9276e-02,\n",
       "            4.2988e-01,  8.6316e-01],\n",
       "          ...,\n",
       "          [ 1.4655e+00,  9.2585e-01,  2.4618e+00,  ..., -6.2093e-01,\n",
       "            6.6788e-01, -1.1132e+00],\n",
       "          [ 7.5520e-01,  9.6412e-01, -2.2369e-02,  ...,  6.3911e-01,\n",
       "            1.1416e+00, -9.4176e-01],\n",
       "          [ 7.7533e-01,  2.2444e-01,  1.6752e+00,  ..., -1.3196e+00,\n",
       "            2.3849e+00,  2.8356e-02]],\n",
       "\n",
       "         [[ 4.8360e-01, -1.0073e+00,  8.0159e-01,  ..., -2.2404e+00,\n",
       "            1.7253e+00,  3.0556e-01],\n",
       "          [ 5.4927e-01,  9.7859e-01,  8.4323e-01,  ..., -6.1060e-01,\n",
       "           -8.0746e-02,  1.4742e+00],\n",
       "          [ 6.9887e-01,  1.3716e+00, -2.3108e-01,  ..., -2.8254e-01,\n",
       "            7.3513e-01, -7.1640e-02],\n",
       "          ...,\n",
       "          [ 4.8723e-01,  5.7511e-01,  2.2183e-04,  ..., -1.2641e+00,\n",
       "           -2.5283e-01,  1.0462e+00],\n",
       "          [ 6.2858e-01,  1.6244e-01,  4.3425e-01,  ..., -1.6021e+00,\n",
       "            5.3575e-03,  1.2416e+00],\n",
       "          [ 8.0946e-02,  2.5442e-01, -7.8181e-01,  ..., -1.2701e+00,\n",
       "            2.5299e+00,  6.9425e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.3213e+00, -1.7498e+00,  2.4595e-01,  ..., -9.1537e-01,\n",
       "           -5.9725e-01, -1.0838e+00],\n",
       "          [ 6.1547e-01,  1.3537e-01, -1.1228e+00,  ..., -1.1072e+00,\n",
       "            7.1802e-01, -9.9031e-01],\n",
       "          [ 1.0063e+00,  9.9588e-01, -1.3076e+00,  ...,  1.9869e-01,\n",
       "            2.6466e-01, -1.0834e+00],\n",
       "          ...,\n",
       "          [-3.4444e-02, -8.3676e-01, -2.6695e-01,  ...,  3.0259e+00,\n",
       "           -2.5068e-02, -1.1373e+00],\n",
       "          [ 1.4231e+00, -2.0475e+00, -9.4141e-01,  ...,  2.0615e+00,\n",
       "           -7.4426e-01, -1.9909e+00],\n",
       "          [ 2.5781e+00, -1.0676e+00,  3.2611e-01,  ..., -2.4092e-01,\n",
       "           -8.7376e-01, -2.7812e+00]],\n",
       "\n",
       "         [[ 1.6083e+00, -4.6607e-01,  1.8621e+00,  ...,  2.2247e-01,\n",
       "            5.0093e-01, -8.8054e-01],\n",
       "          [-2.5039e-01,  4.9782e-01,  1.1330e+00,  ..., -4.4858e-01,\n",
       "            6.8509e-01, -5.9103e-02],\n",
       "          [-3.1313e-01,  5.2445e-01,  7.9307e-01,  ...,  6.2106e-01,\n",
       "            3.6061e-01, -5.8285e-01],\n",
       "          ...,\n",
       "          [ 9.6871e-02, -4.3944e-01,  2.1630e-01,  ..., -1.3637e+00,\n",
       "           -4.8613e-01, -1.0901e+00],\n",
       "          [ 9.0839e-01, -1.1009e+00,  1.9437e-01,  ..., -6.2960e-01,\n",
       "           -1.0295e-01, -1.1893e+00],\n",
       "          [ 1.0664e-01, -1.5591e+00, -2.4004e-01,  ..., -6.4462e-01,\n",
       "            2.7413e-01,  2.6806e-01]],\n",
       "\n",
       "         [[ 1.7282e-01, -7.5419e-01,  4.9264e-01,  ..., -6.8981e-01,\n",
       "            1.6413e+00,  4.7582e-02],\n",
       "          [-2.0229e-01, -5.8297e-02, -1.2249e-02,  ...,  4.8932e-01,\n",
       "           -1.0157e+00,  1.0463e+00],\n",
       "          [-1.5105e+00,  4.9360e-01,  1.1492e+00,  ...,  2.0201e-01,\n",
       "           -1.3328e+00, -7.0610e-01],\n",
       "          ...,\n",
       "          [-1.3470e+00,  1.3836e+00, -3.7882e-01,  ...,  7.4125e-02,\n",
       "           -8.7026e-01,  3.3123e-01],\n",
       "          [-1.3424e+00,  8.0444e-01,  9.7038e-02,  ...,  1.1953e+00,\n",
       "            1.5423e+00, -1.4559e+00],\n",
       "          [ 2.4398e-01,  3.1937e-01,  4.5857e-01,  ...,  8.3211e-01,\n",
       "            3.8265e-01, -8.9117e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.3921e+00, -1.0831e+00, -6.9476e-01,  ...,  4.6400e-01,\n",
       "            2.9854e-01, -7.4609e-01],\n",
       "          [ 1.2384e+00,  8.0104e-01, -1.0602e+00,  ...,  6.3238e-01,\n",
       "           -6.2050e-01, -3.0405e-01],\n",
       "          [-1.6580e-01, -1.9102e-02,  1.0388e-01,  ...,  5.3626e-02,\n",
       "           -1.7731e+00, -3.7017e-01],\n",
       "          ...,\n",
       "          [-1.9989e-01, -1.2490e+00,  1.3260e+00,  ...,  1.3679e-01,\n",
       "            1.3006e+00, -9.8094e-02],\n",
       "          [ 1.0252e+00, -6.3939e-01,  2.0242e+00,  ...,  1.2205e+00,\n",
       "            2.9383e+00,  1.4362e+00],\n",
       "          [ 1.2290e+00, -2.3312e+00,  1.5966e+00,  ...,  1.3119e+00,\n",
       "            7.0447e-01,  2.4694e+00]],\n",
       "\n",
       "         [[ 4.3649e-01,  1.3865e+00,  1.8181e-01,  ..., -5.2774e-01,\n",
       "            1.1624e+00,  5.4379e-01],\n",
       "          [ 4.4061e-02, -4.5399e-01,  1.2858e-01,  ..., -2.4107e+00,\n",
       "           -7.3373e-01,  1.8428e+00],\n",
       "          [ 1.3457e+00, -3.8301e-01, -3.8816e-01,  ..., -1.0561e+00,\n",
       "           -3.7474e-02,  7.2480e-01],\n",
       "          ...,\n",
       "          [ 2.1043e-01, -4.8752e-01,  6.7383e-01,  ..., -7.7413e-01,\n",
       "           -4.7962e-01, -2.6470e-01],\n",
       "          [ 1.5966e+00,  2.1986e+00,  2.9631e+00,  ..., -1.5839e+00,\n",
       "            4.5640e-01, -8.2627e-01],\n",
       "          [ 1.6795e+00,  1.3376e+00,  2.5818e+00,  ..., -1.3017e+00,\n",
       "            1.5489e+00, -1.1731e+00]],\n",
       "\n",
       "         [[ 4.3573e-01, -9.7381e-01,  8.0442e-01,  ..., -2.2530e+00,\n",
       "            1.7684e+00,  3.6865e-01],\n",
       "          [-2.2418e-01, -4.7616e-01,  5.3209e-01,  ..., -1.8659e+00,\n",
       "            8.6255e-01,  1.2825e+00],\n",
       "          [-3.1958e-01, -2.2053e-01, -1.2013e+00,  ..., -3.7495e-01,\n",
       "            2.0621e+00,  3.1856e-03],\n",
       "          ...,\n",
       "          [-3.2663e-01,  6.5410e-01, -6.2113e-01,  ..., -1.4350e+00,\n",
       "            1.2513e+00,  1.4074e+00],\n",
       "          [ 7.5869e-01,  4.4638e-01,  1.3396e+00,  ..., -1.8409e+00,\n",
       "           -6.6144e-01,  6.3147e-01],\n",
       "          [ 1.2785e+00,  1.6497e+00,  2.2093e-01,  ..., -2.5978e+00,\n",
       "            3.9304e-01,  7.9892e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.3330e+00, -1.7258e+00,  2.3756e-01,  ..., -9.7754e-01,\n",
       "           -5.9860e-01, -1.0789e+00],\n",
       "          [ 4.1391e-01, -6.9208e-01, -9.0242e-01,  ..., -4.9802e-02,\n",
       "           -3.5577e-01, -1.1349e+00],\n",
       "          [ 8.5243e-01,  4.3083e-01, -9.2595e-01,  ..., -1.5804e+00,\n",
       "            1.4239e-01, -2.5446e+00],\n",
       "          ...,\n",
       "          [ 9.1563e-01, -7.4117e-01, -3.6849e-01,  ..., -2.6125e-01,\n",
       "            5.0763e-01, -8.7361e-01],\n",
       "          [ 1.6760e+00, -1.4111e+00, -4.5818e-02,  ...,  2.3301e-01,\n",
       "           -6.5377e-01, -2.1563e+00],\n",
       "          [ 1.4833e+00, -1.0417e+00,  6.7740e-01,  ..., -1.0268e-01,\n",
       "           -1.3954e+00, -1.9965e+00]],\n",
       "\n",
       "         [[ 1.6295e+00, -4.7261e-01,  1.8812e+00,  ...,  2.1549e-01,\n",
       "            5.2226e-01, -8.7412e-01],\n",
       "          [ 5.2327e-01, -9.7449e-01,  1.4045e+00,  ...,  5.2581e-01,\n",
       "            1.3695e+00, -3.7822e-01],\n",
       "          [ 3.0306e-01, -1.4166e+00,  4.6785e-01,  ...,  9.6090e-01,\n",
       "            1.2257e-01, -5.2386e-01],\n",
       "          ...,\n",
       "          [ 1.5098e+00,  3.5764e-02,  7.1094e-01,  ...,  1.6715e-01,\n",
       "            1.2272e+00,  1.0038e-01],\n",
       "          [ 4.9342e-01, -2.0323e+00,  7.2095e-01,  ..., -5.0957e-01,\n",
       "            1.2339e+00, -1.2526e+00],\n",
       "          [ 7.8924e-01, -1.1523e+00,  2.1176e-02,  ...,  5.6408e-01,\n",
       "           -3.6161e-02,  4.9863e-01]],\n",
       "\n",
       "         [[ 2.0604e-01, -7.9677e-01,  4.8627e-01,  ..., -7.1315e-01,\n",
       "            1.6499e+00,  8.5130e-02],\n",
       "          [-1.0157e+00,  3.6574e-01,  2.7317e-01,  ...,  9.4251e-01,\n",
       "           -1.5201e+00,  3.5555e-01],\n",
       "          [-5.6452e-01, -6.1293e-02,  4.8700e-01,  ...,  4.0041e-01,\n",
       "           -1.0406e+00,  7.8079e-01],\n",
       "          ...,\n",
       "          [-1.1593e+00,  3.7706e-01, -6.0979e-01,  ...,  1.3281e+00,\n",
       "            1.2064e-02, -1.0601e-01],\n",
       "          [-1.6393e+00,  5.1516e-01,  1.0013e+00,  ...,  1.2647e+00,\n",
       "           -1.5766e-02,  6.0456e-02],\n",
       "          [ 8.3083e-01,  1.1440e+00, -5.5923e-01,  ...,  6.3422e-01,\n",
       "            9.0329e-02,  2.1166e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.4129e+00, -1.0558e+00, -6.7527e-01,  ...,  4.6655e-01,\n",
       "            2.9956e-01, -7.6531e-01],\n",
       "          [ 1.2940e+00,  7.4876e-01,  5.7083e-01,  ..., -1.2930e+00,\n",
       "           -6.3923e-01, -1.7243e-01],\n",
       "          [ 1.0467e+00,  6.3199e-01, -5.4646e-01,  ..., -9.4072e-01,\n",
       "           -3.9978e-01,  1.3386e+00],\n",
       "          ...,\n",
       "          [ 1.2014e-01,  2.3500e-01,  1.1725e+00,  ..., -5.3968e-01,\n",
       "            3.1930e-01,  5.9586e-01],\n",
       "          [ 1.4031e+00, -9.7007e-01,  1.2631e+00,  ...,  5.6367e-01,\n",
       "            1.7878e+00,  2.6563e+00],\n",
       "          [ 4.3835e-01, -1.7442e+00,  1.3000e+00,  ...,  9.0706e-01,\n",
       "            1.2500e+00,  1.6673e+00]],\n",
       "\n",
       "         [[ 4.2410e-01,  1.3648e+00,  1.5210e-01,  ..., -5.0652e-01,\n",
       "            1.1638e+00,  5.4896e-01],\n",
       "          [-2.2513e-01, -3.5099e-01, -6.9436e-01,  ..., -4.5811e-01,\n",
       "            2.2576e-01,  3.7059e-01],\n",
       "          [-7.4099e-01, -2.2256e-01, -1.1621e+00,  ..., -1.3774e+00,\n",
       "            1.5072e+00,  3.7856e-01],\n",
       "          ...,\n",
       "          [ 1.9290e+00,  2.8943e-01, -7.5388e-02,  ..., -6.6136e-01,\n",
       "            7.4275e-01, -7.5233e-01],\n",
       "          [ 1.3111e+00,  1.2647e+00,  1.6157e+00,  ...,  6.6238e-02,\n",
       "            9.0668e-01, -9.0549e-01],\n",
       "          [ 2.6009e+00,  1.2323e+00,  1.9495e+00,  ..., -1.0100e+00,\n",
       "            2.2750e+00, -1.5426e+00]],\n",
       "\n",
       "         [[ 4.6189e-01, -9.7181e-01,  8.0229e-01,  ..., -2.2375e+00,\n",
       "            1.7616e+00,  3.4004e-01],\n",
       "          [ 4.8421e-01, -8.2850e-01, -1.0800e+00,  ..., -3.4660e-01,\n",
       "            6.9403e-01,  5.2055e-01],\n",
       "          [ 6.5496e-01, -1.1746e+00, -7.3517e-01,  ..., -7.7998e-01,\n",
       "            5.4596e-01, -9.9585e-01],\n",
       "          ...,\n",
       "          [ 5.8563e-01, -6.4165e-01,  7.5925e-01,  ..., -1.5009e+00,\n",
       "            7.4119e-01, -1.8486e-01],\n",
       "          [ 9.6604e-01,  6.8299e-01,  1.1308e+00,  ..., -1.3800e+00,\n",
       "           -9.1876e-01, -7.8706e-02],\n",
       "          [ 3.7903e-02,  1.0639e+00, -4.7159e-01,  ..., -1.5327e+00,\n",
       "            8.5440e-01,  1.0405e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.2204e+00,  9.1249e-01,  3.2414e-01,  ..., -7.9780e-01,\n",
       "            1.4206e+00,  1.3262e-01],\n",
       "          [-7.6457e-01, -3.8772e-01, -4.2202e-03,  ...,  9.9277e-02,\n",
       "           -5.2210e-01, -2.4414e-01],\n",
       "          [-8.6259e-01, -2.7835e-03,  3.7261e-01,  ..., -8.2788e-01,\n",
       "           -2.9575e-02,  3.5824e-01],\n",
       "          ...,\n",
       "          [-1.0022e+00, -1.4412e-01,  4.5502e-01,  ..., -8.0358e-02,\n",
       "            5.7039e-01,  8.4446e-01],\n",
       "          [-8.2893e-01, -3.4582e-01,  5.9847e-01,  ...,  3.4734e-01,\n",
       "            5.9480e-02,  3.8985e-01],\n",
       "          [-3.4340e-01, -6.7738e-01,  5.3862e-01,  ...,  5.4811e-01,\n",
       "           -4.5284e-01,  7.9253e-01]],\n",
       "\n",
       "         [[ 1.8090e-01, -1.9198e-01, -1.3832e+00,  ...,  8.9934e-01,\n",
       "            3.6674e-01, -4.1900e-01],\n",
       "          [ 3.0756e-01, -3.8650e-01, -2.7884e-02,  ..., -2.3116e-01,\n",
       "            8.9767e-01, -2.3620e-01],\n",
       "          [ 2.1873e-01, -6.6141e-02, -2.7700e-01,  ...,  2.8938e-01,\n",
       "            4.3176e-01, -7.0035e-01],\n",
       "          ...,\n",
       "          [-2.8449e-01,  3.7231e-01,  1.1818e+00,  ..., -3.1224e-04,\n",
       "            3.8642e-01,  1.1791e+00],\n",
       "          [-1.8799e-01,  2.5556e-01,  4.1316e-01,  ..., -3.0942e-01,\n",
       "           -5.4836e-01,  8.5873e-02],\n",
       "          [-5.5788e-01, -1.3114e+00,  1.0751e+00,  ...,  1.0063e+00,\n",
       "            2.7130e-01, -7.2597e-01]],\n",
       "\n",
       "         [[ 7.7608e-01, -1.1298e-02, -2.3446e-01,  ..., -7.9977e-02,\n",
       "            4.6738e-02, -6.1891e-02],\n",
       "          [-3.1785e-01, -6.7753e-01, -3.7543e-01,  ..., -7.1172e-01,\n",
       "            5.3586e-01,  4.3656e-01],\n",
       "          [-1.7366e-01,  2.5200e-01, -5.4557e-01,  ..., -5.3239e-01,\n",
       "            2.7495e-01,  7.6904e-01],\n",
       "          ...,\n",
       "          [-5.5766e-01, -9.7614e-02, -4.6085e-01,  ...,  7.1640e-01,\n",
       "           -1.2662e+00,  4.3470e-01],\n",
       "          [ 1.0696e+00,  4.1843e-02,  3.0296e-01,  ..., -2.1730e-01,\n",
       "           -1.0123e+00, -1.2323e-01],\n",
       "          [-4.5690e-01,  3.6136e-01, -2.4605e-01,  ..., -1.9561e+00,\n",
       "           -1.3574e+00, -1.5701e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.0246e-01, -5.2339e-01,  5.1443e-01,  ..., -1.3494e+00,\n",
       "            4.0319e-01, -1.8150e-01],\n",
       "          [ 2.0111e-01, -6.8067e-03, -5.0880e-01,  ..., -1.1502e+00,\n",
       "            1.0534e+00,  1.0932e-01],\n",
       "          [ 3.4994e-01, -6.1336e-01, -2.6077e-02,  ..., -6.7296e-01,\n",
       "            5.3926e-01,  7.9579e-01],\n",
       "          ...,\n",
       "          [-1.0938e-01, -4.7171e-01, -2.9361e-02,  ..., -8.5210e-01,\n",
       "            9.1791e-02, -7.6875e-01],\n",
       "          [-6.2101e-02,  6.1418e-01,  1.5733e+00,  ..., -1.5136e+00,\n",
       "            1.5866e-01, -3.0483e-01],\n",
       "          [-8.4820e-01,  3.0726e-01,  1.3572e+00,  ..., -2.4269e+00,\n",
       "           -9.0952e-02,  1.9866e+00]],\n",
       "\n",
       "         [[-6.2576e-01, -4.7126e-02, -1.9427e-01,  ..., -2.5809e-01,\n",
       "            7.1806e-01,  1.7222e-01],\n",
       "          [ 6.6679e-01,  4.4396e-01, -1.6987e-01,  ..., -3.0552e-01,\n",
       "            5.0540e-01,  4.0594e-01],\n",
       "          [-7.7290e-02,  8.8667e-01,  3.7013e-01,  ..., -7.2281e-01,\n",
       "            4.2695e-01, -3.0514e-01],\n",
       "          ...,\n",
       "          [-5.3705e-01,  3.6341e-01,  4.2200e-01,  ..., -2.5590e-01,\n",
       "           -1.1310e+00,  3.2317e-01],\n",
       "          [-3.3367e-01,  3.2436e-02, -1.4102e-01,  ..., -1.0718e+00,\n",
       "            2.1494e-01,  5.3680e-01],\n",
       "          [-1.5221e+00, -1.4147e+00,  2.0333e+00,  ...,  1.4639e+00,\n",
       "            1.0646e+00, -1.4255e+00]],\n",
       "\n",
       "         [[-8.3801e-02, -1.3151e-01, -5.3492e-01,  ..., -2.1309e-02,\n",
       "           -4.9949e-01, -4.0341e-01],\n",
       "          [ 4.5083e-02, -5.6666e-01,  2.2530e+00,  ..., -4.1185e-02,\n",
       "            3.6534e-01,  3.2406e-01],\n",
       "          [ 4.4192e-01, -6.0643e-01, -6.9353e-02,  ...,  1.1727e+00,\n",
       "           -7.5165e-01, -2.0496e-01],\n",
       "          ...,\n",
       "          [-7.6420e-01, -6.9609e-01, -1.5176e+00,  ...,  4.8946e-01,\n",
       "           -6.6997e-01, -7.0969e-01],\n",
       "          [-2.9651e-01,  9.6576e-01, -4.1637e-01,  ...,  1.6821e-01,\n",
       "           -4.4894e-01,  2.1305e-01],\n",
       "          [ 1.3796e+00, -1.0654e+00, -4.2171e-01,  ...,  9.7759e-01,\n",
       "            3.6443e-01, -5.1445e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.2613e+00,  9.3913e-01,  3.5251e-01,  ..., -8.1105e-01,\n",
       "            1.4164e+00,  1.2228e-01],\n",
       "          [ 1.4197e+00, -9.5199e-02, -3.6836e-01,  ..., -4.3561e-01,\n",
       "           -1.3356e+00,  1.2367e-01],\n",
       "          [-6.0996e-02,  1.9621e-01,  6.3789e-01,  ..., -7.2415e-01,\n",
       "            2.1885e-01, -7.5861e-01],\n",
       "          ...,\n",
       "          [-1.3019e-01,  2.3811e-01,  6.5354e-01,  ..., -4.2681e-01,\n",
       "            4.4866e-01, -1.9800e-01],\n",
       "          [-6.8469e-03, -2.2143e-01,  9.3159e-01,  ...,  6.5535e-02,\n",
       "            6.0815e-01,  6.9817e-01],\n",
       "          [-7.8935e-01, -1.0610e-01,  1.5314e+00,  ..., -1.4587e-01,\n",
       "           -2.1171e-01,  3.5986e-01]],\n",
       "\n",
       "         [[ 1.6337e-01, -1.9127e-01, -1.3916e+00,  ...,  9.2723e-01,\n",
       "            3.4716e-01, -4.1861e-01],\n",
       "          [ 9.6740e-01,  1.1367e-01,  1.2668e-01,  ..., -4.2477e-01,\n",
       "           -2.1288e-01, -1.4377e-02],\n",
       "          [ 5.2214e-01, -7.3439e-01,  5.8152e-01,  ..., -5.3720e-02,\n",
       "            1.5346e-01, -8.6034e-01],\n",
       "          ...,\n",
       "          [-3.1287e-01,  2.7483e-01,  1.0096e-01,  ...,  7.5606e-01,\n",
       "            7.8745e-01,  7.7812e-01],\n",
       "          [ 5.7939e-02, -4.5569e-01, -1.0037e-01,  ...,  3.9528e-01,\n",
       "            9.5659e-01,  3.3772e-02],\n",
       "          [-8.4456e-01, -3.5242e-01,  5.0537e-01,  ...,  1.4907e+00,\n",
       "            7.3411e-01, -3.0972e-01]],\n",
       "\n",
       "         [[ 7.9676e-01,  6.3875e-03, -2.5108e-01,  ..., -9.1211e-02,\n",
       "            5.8880e-02, -8.3453e-02],\n",
       "          [-5.1699e-01, -5.6295e-01, -2.5153e-01,  ..., -2.9502e-01,\n",
       "            5.8315e-01,  1.0265e+00],\n",
       "          [-1.0701e+00, -4.1805e-01, -2.6658e-01,  ..., -5.9586e-01,\n",
       "           -5.1556e-01,  7.8786e-01],\n",
       "          ...,\n",
       "          [-1.1324e+00,  3.6021e-01,  2.0779e-01,  ...,  1.3704e-02,\n",
       "           -6.1445e-02, -3.5067e-01],\n",
       "          [-6.0272e-01,  3.4712e-02,  1.5149e-01,  ...,  5.8609e-01,\n",
       "           -2.4208e-01,  2.6747e-01],\n",
       "          [-4.6452e-01,  5.5879e-01, -1.0529e-01,  ..., -8.1819e-01,\n",
       "           -8.7708e-01, -8.2836e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.2136e-01, -4.8955e-01,  4.6856e-01,  ..., -1.3532e+00,\n",
       "            4.0559e-01, -1.3468e-01],\n",
       "          [ 3.5165e-01, -1.4480e-01, -3.4037e-01,  ..., -4.5330e-01,\n",
       "            2.8010e-01,  1.3728e+00],\n",
       "          [-6.5127e-04, -1.1045e+00,  3.3414e-01,  ..., -1.2339e+00,\n",
       "            7.7488e-01,  2.9664e-01],\n",
       "          ...,\n",
       "          [-4.6529e-01, -4.8350e-01,  3.5415e-02,  ..., -1.3403e+00,\n",
       "           -5.5314e-01,  2.9828e-01],\n",
       "          [-4.1712e-02,  5.3990e-01, -1.2647e-01,  ..., -7.7138e-01,\n",
       "           -1.8449e-02,  6.8935e-01],\n",
       "          [-9.8744e-01,  7.2486e-01, -2.9505e-01,  ..., -1.1791e+00,\n",
       "            6.9512e-01,  4.2667e-01]],\n",
       "\n",
       "         [[-6.2288e-01, -1.8955e-02, -1.8790e-01,  ..., -2.7352e-01,\n",
       "            7.1350e-01,  1.2134e-01],\n",
       "          [ 4.4854e-01, -2.1349e-01,  1.2624e-01,  ...,  6.4082e-01,\n",
       "           -5.6265e-01, -4.6908e-01],\n",
       "          [-4.2759e-01,  1.7221e-01, -3.2465e-01,  ...,  2.7208e-01,\n",
       "           -5.3060e-01, -3.5447e-01],\n",
       "          ...,\n",
       "          [-5.8311e-01,  2.0382e-01,  8.2098e-01,  ...,  4.5501e-01,\n",
       "           -2.3428e-01, -7.8781e-01],\n",
       "          [-5.5177e-01, -1.0896e-01,  6.7195e-01,  ..., -7.9770e-01,\n",
       "           -3.0047e-02,  7.1236e-02],\n",
       "          [-1.6745e+00, -1.3967e+00,  1.0694e+00,  ..., -3.1486e-01,\n",
       "            1.5501e+00, -1.2765e+00]],\n",
       "\n",
       "         [[-8.1925e-02, -1.1156e-01, -5.1684e-01,  ..., -1.4945e-02,\n",
       "           -4.3715e-01, -4.2725e-01],\n",
       "          [-1.4882e-01, -2.0546e-01,  1.3324e-01,  ..., -7.9420e-01,\n",
       "            8.1696e-01, -1.5183e-01],\n",
       "          [-1.1736e-01, -3.3134e-01, -1.4304e-01,  ...,  6.7222e-01,\n",
       "           -8.5390e-01,  1.8095e-01],\n",
       "          ...,\n",
       "          [-9.5817e-01, -8.5239e-01,  1.5186e-01,  ..., -4.7886e-01,\n",
       "           -4.5439e-01, -1.4405e+00],\n",
       "          [-1.8322e-01, -7.5849e-02, -1.0431e+00,  ...,  7.8924e-01,\n",
       "           -1.0652e+00,  1.5465e-01],\n",
       "          [ 3.0408e-01, -3.7996e-01, -1.3452e+00,  ...,  5.1315e-01,\n",
       "           -1.3234e-01, -1.3310e+00]]],\n",
       "\n",
       "\n",
       "        [[[-1.2553e+00,  9.3649e-01,  3.3600e-01,  ..., -7.9576e-01,\n",
       "            1.4224e+00,  1.2051e-01],\n",
       "          [-1.6773e+00,  9.7214e-01,  7.1359e-01,  ..., -2.9672e-01,\n",
       "            7.2078e-01, -1.9555e-01],\n",
       "          [-8.3914e-01,  7.6454e-01,  6.9072e-01,  ..., -2.1344e-01,\n",
       "            8.1010e-01, -2.9539e-01],\n",
       "          ...,\n",
       "          [-1.3114e-01,  8.6187e-01,  3.0145e-01,  ..., -5.6875e-01,\n",
       "            2.6024e-01,  6.5826e-01],\n",
       "          [-6.7178e-01,  1.8415e-01,  2.1401e-01,  ...,  4.7238e-02,\n",
       "            2.1806e-01,  4.5893e-01],\n",
       "          [-1.1816e+00, -4.0859e-01,  1.9949e+00,  ...,  7.7664e-02,\n",
       "            4.9763e-02,  2.5034e-01]],\n",
       "\n",
       "         [[ 1.8765e-01, -1.9579e-01, -1.3820e+00,  ...,  9.3786e-01,\n",
       "            3.4764e-01, -4.5219e-01],\n",
       "          [-5.0924e-02, -3.2857e-01,  3.7607e-01,  ...,  8.2667e-01,\n",
       "            3.1308e-01,  1.5881e-01],\n",
       "          [ 1.8210e-02, -1.0447e+00, -1.6201e-01,  ...,  6.3384e-01,\n",
       "            3.4230e-01, -8.6000e-01],\n",
       "          ...,\n",
       "          [ 4.3318e-01, -2.6852e-01, -1.5021e-01,  ..., -1.2080e-01,\n",
       "           -5.1128e-02, -2.8837e-01],\n",
       "          [ 4.8863e-01, -6.9883e-01,  4.8281e-04,  ...,  5.0839e-03,\n",
       "           -1.4563e-01, -2.1954e-01],\n",
       "          [-9.5386e-01, -1.4348e-01,  6.4714e-01,  ...,  1.0672e+00,\n",
       "            4.4275e-01, -5.2284e-01]],\n",
       "\n",
       "         [[ 7.9984e-01, -4.8503e-03, -2.3645e-01,  ..., -9.0414e-02,\n",
       "            7.6721e-02, -5.2975e-02],\n",
       "          [ 6.9707e-01,  8.2372e-02, -4.7570e-01,  ..., -5.7080e-01,\n",
       "           -1.0648e+00,  5.3770e-01],\n",
       "          [ 3.2337e-01, -5.9699e-02, -5.3896e-01,  ..., -7.9377e-01,\n",
       "           -1.4202e+00,  2.2666e-01],\n",
       "          ...,\n",
       "          [-1.9726e-01, -5.3540e-02, -5.9182e-01,  ..., -3.4835e-01,\n",
       "           -1.0427e+00, -2.8887e-01],\n",
       "          [ 9.4357e-01,  1.0421e+00, -4.1325e-01,  ...,  3.5179e-01,\n",
       "           -1.0230e-01, -7.7413e-01],\n",
       "          [-7.4453e-01,  5.4274e-01, -1.4337e+00,  ..., -7.4196e-01,\n",
       "           -1.9144e+00, -7.4974e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.3972e-01, -4.7881e-01,  4.7520e-01,  ..., -1.3480e+00,\n",
       "            4.2410e-01, -1.4020e-01],\n",
       "          [-5.4216e-01,  4.5352e-01,  1.0919e+00,  ..., -1.8723e+00,\n",
       "           -9.5114e-02,  1.6605e-01],\n",
       "          [-7.6513e-01,  9.1102e-01,  1.4154e+00,  ..., -1.8154e+00,\n",
       "            3.9640e-01,  4.0055e-01],\n",
       "          ...,\n",
       "          [-9.1365e-01,  1.0138e-01,  1.2371e+00,  ..., -9.0869e-01,\n",
       "            2.1374e-01,  5.2377e-02],\n",
       "          [-6.2065e-01,  5.2917e-01,  1.2894e+00,  ..., -1.3155e+00,\n",
       "            7.3728e-02,  1.4322e-01],\n",
       "          [-2.0640e+00,  2.1285e-01,  1.3193e+00,  ..., -1.4458e+00,\n",
       "           -1.1270e-01, -2.2547e-02]],\n",
       "\n",
       "         [[-6.3081e-01, -6.0667e-02, -2.0818e-01,  ..., -2.6141e-01,\n",
       "            7.1207e-01,  1.4316e-01],\n",
       "          [-4.6341e-01,  4.4363e-01,  4.7559e-01,  ..., -1.1200e+00,\n",
       "           -4.2843e-01, -4.0626e-01],\n",
       "          [-1.1536e+00, -3.0823e-01,  7.0377e-01,  ..., -1.2201e+00,\n",
       "            5.6682e-01, -6.6712e-01],\n",
       "          ...,\n",
       "          [-4.3967e-01,  3.4325e-01, -4.0354e-01,  ..., -8.9099e-02,\n",
       "           -2.4508e-01, -3.7342e-01],\n",
       "          [-3.8760e-03, -6.5728e-01, -3.9988e-01,  ..., -1.1010e+00,\n",
       "            8.1945e-01,  3.6051e-01],\n",
       "          [-2.4965e+00, -3.9517e-01,  1.0363e+00,  ...,  5.8428e-01,\n",
       "            5.1573e-01, -8.2065e-01]],\n",
       "\n",
       "         [[-7.0911e-02, -1.3178e-01, -5.1816e-01,  ..., -3.6116e-02,\n",
       "           -4.6031e-01, -4.2041e-01],\n",
       "          [-1.3838e+00, -2.4779e-01, -3.1428e-01,  ..., -3.4930e-01,\n",
       "           -1.5188e+00, -4.3313e-01],\n",
       "          [-1.0039e+00, -5.6671e-01, -9.8739e-01,  ...,  1.6531e-01,\n",
       "           -1.6227e+00, -6.9291e-01],\n",
       "          ...,\n",
       "          [-1.2870e+00, -5.8904e-01,  1.3182e+00,  ..., -2.6447e-01,\n",
       "           -4.1757e-01, -4.8534e-01],\n",
       "          [-8.0788e-02,  5.6138e-01, -1.6447e+00,  ...,  7.9451e-01,\n",
       "           -1.1407e+00, -7.0761e-01],\n",
       "          [-1.7002e-01, -1.5441e-01, -9.2266e-01,  ..., -5.4489e-01,\n",
       "           -5.1281e-01, -1.0593e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 1.4224e+00,  3.7903e-01, -3.0389e+00,  ...,  2.3953e+00,\n",
       "            1.0900e+00, -2.3439e+00],\n",
       "          [-4.1240e-01,  2.9750e-01,  2.0281e-01,  ...,  9.2738e-01,\n",
       "            1.3487e+00,  1.2870e+00],\n",
       "          [-1.4988e+00,  2.3874e+00, -2.8128e+00,  ...,  2.2919e+00,\n",
       "            2.3655e+00,  5.4079e+00],\n",
       "          ...,\n",
       "          [ 1.2664e-01,  2.8692e+00,  3.3258e-01,  ...,  5.3862e-01,\n",
       "           -1.0923e+00,  2.9987e+00],\n",
       "          [ 1.4902e+00,  6.0936e-01, -2.0325e-04,  ...,  1.8475e+00,\n",
       "            2.8454e+00,  4.3535e+00],\n",
       "          [ 7.1992e-01,  4.5758e-01, -9.3374e-01,  ..., -1.0655e+00,\n",
       "            1.1692e+00,  1.8114e+00]],\n",
       "\n",
       "         [[ 3.1051e+00,  1.8593e+00, -9.2833e-01,  ..., -2.1398e-01,\n",
       "            2.9975e+00,  1.2471e+00],\n",
       "          [ 6.3895e-01,  2.7830e+00, -1.1666e+00,  ..., -6.1706e-01,\n",
       "            2.1181e+00,  2.1242e+00],\n",
       "          [-2.8169e-01,  2.3800e+00,  2.0715e-01,  ..., -1.8860e+00,\n",
       "            2.6473e+00,  2.9405e+00],\n",
       "          ...,\n",
       "          [ 2.8034e-01,  1.1912e+00,  1.1874e+00,  ...,  7.5410e-01,\n",
       "            4.3328e+00,  9.7448e-01],\n",
       "          [-2.8999e+00, -1.5761e+00, -3.1496e+00,  ..., -4.1152e-01,\n",
       "            2.8700e+00,  1.1694e+00],\n",
       "          [ 1.3067e+00, -2.8400e+00, -1.0245e-01,  ..., -7.9293e-01,\n",
       "           -5.0296e-01, -2.4340e-01]],\n",
       "\n",
       "         [[-8.0321e-01, -3.3868e-01,  3.2218e+00,  ..., -2.1272e-01,\n",
       "           -2.0831e+00, -7.7023e-01],\n",
       "          [ 9.3314e-01,  1.6733e-01,  4.5767e+00,  ..., -2.2133e+00,\n",
       "           -3.6181e+00, -8.8402e-01],\n",
       "          [ 2.9699e+00, -5.0481e-02,  1.5375e+00,  ..., -3.5346e+00,\n",
       "           -2.1754e+00, -1.5260e+00],\n",
       "          ...,\n",
       "          [ 2.0696e+00, -2.3190e+00,  5.0663e-01,  ..., -2.8309e+00,\n",
       "           -6.7691e-01, -5.3794e+00],\n",
       "          [ 3.5855e+00, -2.2205e+00,  1.0795e+00,  ...,  1.6056e+00,\n",
       "           -3.0606e+00, -1.2771e+00],\n",
       "          [-3.0470e+00,  2.3321e+00,  1.9023e+00,  ...,  1.8180e+00,\n",
       "           -5.1015e-01,  5.7467e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.4263e+00, -8.2843e-01,  3.2262e+00,  ..., -1.7133e+00,\n",
       "           -1.5669e+00,  3.1231e+00],\n",
       "          [ 5.3324e-01,  1.0940e+00, -1.2444e+00,  ..., -2.1453e+00,\n",
       "           -4.8201e+00,  1.3661e+00],\n",
       "          [-1.5532e+00, -6.2152e-02, -1.8668e+00,  ...,  4.2593e-01,\n",
       "           -1.2017e+00,  2.0366e+00],\n",
       "          ...,\n",
       "          [-1.9400e+00,  1.6232e+00, -7.9068e-01,  ..., -8.7383e-01,\n",
       "           -1.6876e+00,  2.7633e+00],\n",
       "          [-3.8863e-01, -1.8419e-01, -3.1729e-01,  ..., -1.2808e+00,\n",
       "           -2.3614e+00,  3.9316e-01],\n",
       "          [-5.4556e-02, -8.2356e-01, -3.8270e-01,  ...,  5.3508e-01,\n",
       "           -2.2894e+00, -1.8350e+00]],\n",
       "\n",
       "         [[-2.5671e+00, -2.4231e-01,  1.2058e-01,  ...,  1.3228e+00,\n",
       "           -7.7082e-01,  1.5601e+00],\n",
       "          [-5.3928e+00, -2.0877e+00,  2.7420e+00,  ..., -5.6272e-02,\n",
       "            1.7908e+00, -6.4915e-01],\n",
       "          [-2.5719e+00, -2.5733e+00,  1.4008e+00,  ..., -3.5323e-01,\n",
       "            3.3517e+00, -1.7854e+00],\n",
       "          ...,\n",
       "          [-3.0471e+00, -6.0167e-01, -1.2963e+00,  ...,  3.8177e-01,\n",
       "            1.2815e+00,  1.2720e+00],\n",
       "          [-2.3676e+00,  6.8823e-01, -3.0554e-01,  ...,  2.1355e-02,\n",
       "           -6.6120e-01, -2.7560e+00],\n",
       "          [-1.4364e-01,  2.7376e+00, -4.2380e+00,  ...,  1.7266e+00,\n",
       "           -5.7626e+00, -2.7016e+00]],\n",
       "\n",
       "         [[-1.6136e+00, -3.1827e+00, -4.0736e-01,  ...,  4.1827e-01,\n",
       "           -5.9471e-01,  1.8030e+00],\n",
       "          [-6.4273e-01, -2.4127e+00,  3.3408e-01,  ...,  3.3365e+00,\n",
       "           -2.4501e+00,  3.0126e+00],\n",
       "          [-5.6385e+00,  7.2435e-01, -7.6741e-01,  ...,  1.8652e+00,\n",
       "            2.1415e+00,  2.5765e+00],\n",
       "          ...,\n",
       "          [-6.3752e-01, -2.4057e+00,  3.5480e+00,  ...,  2.8464e+00,\n",
       "            2.3581e+00,  1.7696e+00],\n",
       "          [ 1.0379e+00, -5.6471e-01, -1.5138e+00,  ...,  1.0896e+00,\n",
       "            3.0851e+00,  4.8427e-01],\n",
       "          [-8.2453e-01, -3.1957e+00,  1.3042e+00,  ..., -2.4932e+00,\n",
       "            2.5249e+00,  2.9116e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 2.2196e+00, -3.3861e+00, -2.6668e-01,  ...,  5.5011e-02,\n",
       "           -9.8306e-01, -2.4498e+00],\n",
       "          [ 1.2412e+00,  3.1952e+00, -2.5675e+00,  ...,  3.0741e-01,\n",
       "           -3.3073e-01,  3.8303e+00],\n",
       "          [-9.0794e-02, -8.1185e-01,  1.6191e+00,  ..., -1.3612e+00,\n",
       "            1.2465e+00,  4.8144e-01],\n",
       "          ...,\n",
       "          [-1.7764e+00,  8.1635e-01, -2.1706e-01,  ...,  3.7223e+00,\n",
       "            5.0008e+00, -1.8320e+00],\n",
       "          [-5.5904e-01,  3.1655e-01, -6.1752e-01,  ...,  1.9283e+00,\n",
       "            3.3890e+00,  7.9629e-02],\n",
       "          [ 6.3412e-01,  4.5747e-01, -9.3729e-01,  ..., -9.5723e-01,\n",
       "            1.2401e+00,  1.8262e+00]],\n",
       "\n",
       "         [[ 2.3973e+00,  2.4635e+00, -2.1418e+00,  ...,  2.6466e+00,\n",
       "            1.9245e+00,  6.4687e-01],\n",
       "          [ 2.8423e-01,  1.7023e+00, -7.5501e-02,  ..., -1.0298e+00,\n",
       "            1.8818e+00,  3.0999e+00],\n",
       "          [-9.5746e-01,  2.2092e+00, -1.4794e+00,  ...,  2.3109e-01,\n",
       "            1.1413e+00,  5.4711e-01],\n",
       "          ...,\n",
       "          [-4.1439e-01, -1.2374e-01,  3.1829e+00,  ..., -1.9971e+00,\n",
       "            6.2596e+00,  1.6097e-02],\n",
       "          [-1.6777e+00, -1.8820e+00, -1.5115e+00,  ..., -1.0631e+00,\n",
       "            2.1355e+00,  1.0174e+00],\n",
       "          [ 1.4002e+00, -2.8630e+00,  2.8755e-02,  ..., -7.9198e-01,\n",
       "           -4.8889e-01, -1.7357e-01]],\n",
       "\n",
       "         [[ 1.3920e+00,  6.0557e-01,  2.0031e+00,  ..., -2.4382e+00,\n",
       "           -1.0713e-02, -2.8366e+00],\n",
       "          [ 4.2416e+00,  1.2255e+00, -1.7168e+00,  ..., -3.7068e+00,\n",
       "           -2.2071e+00, -1.8179e+00],\n",
       "          [ 1.6380e+00,  1.3432e+00,  1.5815e+00,  ..., -1.2097e+00,\n",
       "           -3.0720e+00, -3.8411e+00],\n",
       "          ...,\n",
       "          [ 2.0807e-01,  3.1240e-01,  4.4491e+00,  ..., -1.2055e-01,\n",
       "           -1.0889e+00, -2.7980e+00],\n",
       "          [ 1.6932e+00, -2.6659e+00,  2.6491e+00,  ..., -9.1182e-01,\n",
       "           -3.4890e+00,  1.1404e+00],\n",
       "          [-3.1026e+00,  2.2889e+00,  1.9020e+00,  ...,  1.7385e+00,\n",
       "           -6.0771e-01,  5.8709e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.5430e-01,  2.9234e+00,  1.0185e+00,  ..., -9.5047e-01,\n",
       "           -2.8685e+00,  3.8834e+00],\n",
       "          [-9.1821e-01,  1.9749e+00, -2.0502e+00,  ..., -7.1188e-02,\n",
       "            1.5536e+00,  3.9879e+00],\n",
       "          [-2.9485e+00,  2.4288e+00, -1.5258e+00,  ...,  7.4255e-01,\n",
       "            1.3630e+00,  4.7459e+00],\n",
       "          ...,\n",
       "          [-2.8908e+00, -9.2260e-01, -2.3270e-01,  ..., -1.5434e+00,\n",
       "           -2.4242e+00,  2.8059e+00],\n",
       "          [-5.6316e-02, -6.1448e-03,  2.9338e-02,  ..., -2.6394e+00,\n",
       "           -4.0057e+00,  1.2433e+00],\n",
       "          [-9.3397e-02, -9.3440e-01, -4.3864e-01,  ...,  3.9188e-01,\n",
       "           -2.3129e+00, -1.8124e+00]],\n",
       "\n",
       "         [[-1.4900e+00,  1.8951e+00, -8.8788e-01,  ..., -9.7286e-01,\n",
       "           -1.7847e+00, -2.5559e-01],\n",
       "          [-8.6274e-01, -4.7966e-01, -2.0008e-01,  ..., -1.4876e+00,\n",
       "            2.7208e+00, -1.1466e+00],\n",
       "          [-6.3304e-01, -3.4304e+00, -2.3425e-01,  ...,  4.3520e-01,\n",
       "           -6.5315e-02, -4.9472e-01],\n",
       "          ...,\n",
       "          [-1.6978e+00, -1.2434e+00,  1.2706e+00,  ...,  5.6861e+00,\n",
       "           -2.5201e+00,  1.9041e+00],\n",
       "          [-2.7251e+00, -5.8319e-01,  1.7997e+00,  ...,  3.9977e+00,\n",
       "           -1.0100e+00, -1.3508e+00],\n",
       "          [-4.6962e-02,  2.7315e+00, -4.2023e+00,  ...,  1.9217e+00,\n",
       "           -5.6391e+00, -2.7659e+00]],\n",
       "\n",
       "         [[-3.5558e+00, -2.0968e+00,  1.0950e+00,  ...,  9.1220e-02,\n",
       "           -8.0985e-01,  1.8971e+00],\n",
       "          [-3.6054e+00,  2.1063e+00,  3.9040e-02,  ..., -4.5610e-02,\n",
       "            2.2011e+00, -2.0646e-01],\n",
       "          [-8.3493e-01,  6.3080e-01, -2.0188e+00,  ...,  8.5921e-01,\n",
       "            3.4116e+00,  3.4626e+00],\n",
       "          ...,\n",
       "          [-1.2266e+00,  3.5191e-01, -1.5214e+00,  ..., -5.2991e-01,\n",
       "            3.6579e+00, -6.8458e-01],\n",
       "          [-8.6226e-01, -2.7974e+00, -4.8085e-01,  ..., -1.3411e+00,\n",
       "            4.9217e-01, -1.8367e+00],\n",
       "          [-1.0123e+00, -3.2815e+00,  1.3426e+00,  ..., -2.5098e+00,\n",
       "            2.5235e+00,  2.8872e+00]]],\n",
       "\n",
       "\n",
       "        [[[-1.2540e+00, -1.2821e+00, -2.7987e+00,  ..., -2.5858e-01,\n",
       "            8.1950e-02,  4.6455e-01],\n",
       "          [-2.0408e+00,  1.4093e+00, -5.1973e+00,  ..., -1.1547e+00,\n",
       "            1.3961e+00,  3.2337e+00],\n",
       "          [-2.3934e+00, -8.0781e-01, -2.4646e+00,  ..., -1.1365e-01,\n",
       "            2.6995e-01,  3.0825e+00],\n",
       "          ...,\n",
       "          [-5.2170e-01,  2.2531e+00, -2.3405e+00,  ...,  4.1008e+00,\n",
       "            1.6208e+00,  4.4844e+00],\n",
       "          [ 9.5430e-01,  8.6688e-01,  3.5743e-01,  ...,  1.1623e+00,\n",
       "            3.1190e+00, -6.7962e-02],\n",
       "          [ 6.8768e-01,  4.2674e-01, -8.7074e-01,  ..., -1.1028e+00,\n",
       "            1.2075e+00,  1.8396e+00]],\n",
       "\n",
       "         [[-5.4933e-01,  2.6536e+00, -4.7240e-02,  ...,  2.4732e+00,\n",
       "           -2.5323e+00, -3.6050e-01],\n",
       "          [-1.6490e+00,  3.6646e+00, -3.7339e+00,  ...,  1.5990e+00,\n",
       "           -4.0088e-01, -1.8295e+00],\n",
       "          [-1.7917e+00,  2.6515e+00, -9.0029e-01,  ..., -1.0985e+00,\n",
       "           -1.5505e+00,  1.0056e-01],\n",
       "          ...,\n",
       "          [-1.8696e-01,  1.1255e+00, -5.3216e-01,  ...,  4.6475e-02,\n",
       "            7.8542e-01, -6.3460e-01],\n",
       "          [-2.4669e+00, -1.4814e+00, -3.0225e+00,  ...,  1.0033e+00,\n",
       "            3.2945e+00, -1.0221e+00],\n",
       "          [ 1.3105e+00, -2.8513e+00, -5.4511e-02,  ..., -7.2191e-01,\n",
       "           -5.1707e-01, -2.6305e-01]],\n",
       "\n",
       "         [[ 4.0269e-01, -2.9528e+00,  4.1903e+00,  ..., -1.8727e+00,\n",
       "            6.1670e-01, -1.5377e+00],\n",
       "          [ 3.6888e-01, -4.6558e-01,  2.9908e+00,  ..., -3.3838e+00,\n",
       "            2.9781e-01, -4.7067e+00],\n",
       "          [ 1.0892e+00,  1.3078e-01, -2.2895e+00,  ..., -1.5721e+00,\n",
       "           -1.8727e+00, -1.4891e+00],\n",
       "          ...,\n",
       "          [ 1.6684e+00,  1.5677e+00, -5.2938e-01,  ...,  1.4754e+00,\n",
       "           -6.2581e-01, -5.7852e-01],\n",
       "          [-1.3016e+00, -9.8794e-02,  2.1655e+00,  ...,  1.4836e+00,\n",
       "            1.2718e+00, -1.7629e+00],\n",
       "          [-3.1187e+00,  2.3219e+00,  1.9901e+00,  ...,  1.7343e+00,\n",
       "           -6.0658e-01,  5.2284e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.9801e+00,  1.6453e-01,  8.5550e-01,  ...,  1.6616e+00,\n",
       "           -1.8905e+00,  2.7240e+00],\n",
       "          [ 2.7793e+00,  1.8658e-01,  9.8443e-01,  ..., -1.5857e+00,\n",
       "           -1.8626e-01,  2.0906e+00],\n",
       "          [ 4.2397e+00, -2.9388e+00, -1.1238e+00,  ..., -3.8889e+00,\n",
       "            7.0419e-01,  2.5276e+00],\n",
       "          ...,\n",
       "          [ 1.6927e-01, -1.4323e+00, -1.5872e+00,  ..., -3.7333e+00,\n",
       "           -5.6224e+00, -4.2414e-01],\n",
       "          [ 1.2361e-01, -6.3777e-01,  5.4308e-01,  ..., -5.8340e-01,\n",
       "           -1.9729e+00,  2.9181e+00],\n",
       "          [-2.0084e-02, -8.7117e-01, -3.9082e-01,  ...,  4.6073e-01,\n",
       "           -2.2438e+00, -1.7664e+00]],\n",
       "\n",
       "         [[-6.2762e+00, -1.1368e+00, -7.1001e-02,  ..., -1.6354e+00,\n",
       "            1.9976e+00,  3.8406e-01],\n",
       "          [-4.9790e+00, -4.7622e+00, -1.5624e+00,  ..., -7.9178e-01,\n",
       "            1.4320e+00, -1.1185e+00],\n",
       "          [-2.6533e+00, -1.7880e+00, -3.2850e+00,  ...,  2.5343e+00,\n",
       "            2.2183e+00, -3.4382e+00],\n",
       "          ...,\n",
       "          [-1.9136e-01,  1.2246e+00, -3.1436e+00,  ...,  4.6778e-01,\n",
       "            2.9596e-03, -3.2591e+00],\n",
       "          [-1.5465e+00,  7.2638e-01,  1.0131e+00,  ...,  1.6127e+00,\n",
       "           -2.6596e-01, -3.0931e+00],\n",
       "          [-6.1799e-02,  2.7195e+00, -4.2027e+00,  ...,  1.8199e+00,\n",
       "           -5.8059e+00, -2.7303e+00]],\n",
       "\n",
       "         [[-5.6773e-01, -3.9754e+00,  1.3477e-01,  ..., -2.0215e+00,\n",
       "            1.2556e-01,  2.8944e+00],\n",
       "          [-2.0648e+00, -2.0566e+00, -9.5067e-02,  ...,  2.1508e+00,\n",
       "            3.7019e+00,  2.7295e+00],\n",
       "          [-2.1442e+00,  8.2363e-01,  7.9711e-01,  ..., -3.0603e+00,\n",
       "           -1.8468e+00, -3.3671e+00],\n",
       "          ...,\n",
       "          [-1.3995e+00, -2.9124e+00, -1.9079e-01,  ..., -1.4952e+00,\n",
       "           -1.7942e+00, -1.7244e+00],\n",
       "          [ 3.6506e+00, -1.0367e+00,  5.2002e-01,  ..., -4.7968e-01,\n",
       "            3.8284e+00,  2.1425e+00],\n",
       "          [-8.1850e-01, -3.1737e+00,  1.3920e+00,  ..., -2.5441e+00,\n",
       "            2.5303e+00,  2.8750e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 2.3177e+00,  1.0753e+00, -4.9544e-01,  ..., -9.4858e-01,\n",
       "            9.1929e-01,  1.4948e+00],\n",
       "          [ 5.2439e-01, -5.1082e-02, -3.6284e-01,  ...,  1.8357e+00,\n",
       "            9.1424e-01,  1.1746e+00],\n",
       "          [-1.7673e-01,  8.9439e-01, -7.6600e-01,  ...,  7.2168e-01,\n",
       "           -1.7920e-01,  7.2499e-01],\n",
       "          ...,\n",
       "          [-9.4090e-01, -8.2050e-01, -8.8883e-02,  ...,  1.0343e-01,\n",
       "           -1.6510e+00,  6.6186e-01],\n",
       "          [-6.1734e-01,  8.0808e-01,  5.0333e-01,  ..., -1.7596e-01,\n",
       "           -7.8186e-02,  1.7254e+00],\n",
       "          [ 1.3103e-02,  1.8652e-01,  3.2636e-01,  ..., -9.0722e-02,\n",
       "           -2.2243e-01, -5.5148e-01]],\n",
       "\n",
       "         [[ 1.2491e+00,  6.1822e-02,  5.1036e-01,  ...,  5.4652e-01,\n",
       "           -2.8585e+00, -2.8307e+00],\n",
       "          [-1.2360e+00,  2.5067e-01, -4.2752e-01,  ...,  1.4639e+00,\n",
       "           -3.1002e-01, -2.0068e+00],\n",
       "          [-5.2769e-01, -1.3452e+00, -5.7992e-02,  ...,  2.5356e+00,\n",
       "            1.1983e+00, -1.7753e+00],\n",
       "          ...,\n",
       "          [-5.7382e-01, -5.4654e-01, -3.1760e-01,  ...,  9.6205e-01,\n",
       "            1.5246e+00, -2.1455e+00],\n",
       "          [-1.4874e+00, -6.0755e-01, -4.0671e-01,  ...,  1.1392e+00,\n",
       "            2.7415e+00,  7.4279e-01],\n",
       "          [ 2.2991e-01, -9.5008e-01, -9.1554e-02,  ...,  2.2595e-01,\n",
       "            3.1077e-02,  1.1377e+00]],\n",
       "\n",
       "         [[ 9.3804e-01,  5.2157e-01,  5.6198e-01,  ...,  2.3138e+00,\n",
       "            2.1228e+00, -9.9547e-02],\n",
       "          [-3.6676e-01, -1.0965e+00, -2.5180e-02,  ...,  1.8556e+00,\n",
       "            1.2718e+00,  5.2387e-02],\n",
       "          [-3.6721e-01,  4.1445e-01,  2.6993e+00,  ...,  3.2847e+00,\n",
       "            1.1431e-01,  2.6941e-01],\n",
       "          ...,\n",
       "          [ 5.0841e-01, -4.8752e-01, -1.3489e+00,  ...,  2.2109e+00,\n",
       "            1.0779e+00, -2.4865e-02],\n",
       "          [ 4.7327e-01, -4.5647e-01, -7.8515e-01,  ..., -8.9319e-02,\n",
       "            1.2529e+00,  1.4720e-01],\n",
       "          [ 4.1892e-01,  2.9185e-01, -3.9117e-01,  ..., -6.6996e-01,\n",
       "            6.6580e-01,  1.7679e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-7.3337e-02, -3.0118e+00,  6.6257e-01,  ...,  1.1218e+00,\n",
       "           -1.8299e-01, -1.4564e+00],\n",
       "          [ 2.0927e+00, -2.1767e+00,  7.4515e-01,  ...,  2.3193e+00,\n",
       "            1.3315e+00, -2.4957e+00],\n",
       "          [-1.0611e-01, -1.7596e+00,  1.3795e+00,  ...,  9.0028e-01,\n",
       "           -5.6662e-01, -2.6098e+00],\n",
       "          ...,\n",
       "          [ 2.6179e+00, -7.8882e-01,  3.4206e-01,  ...,  8.4344e-01,\n",
       "           -2.1773e-01, -8.3603e-01],\n",
       "          [ 1.1446e-02, -1.4020e+00,  5.0452e-01,  ...,  6.0564e-01,\n",
       "           -1.7094e-01, -1.5569e+00],\n",
       "          [ 2.8698e-01, -2.2990e-02, -2.3009e-01,  ...,  1.2418e+00,\n",
       "           -8.4073e-01, -5.2246e-01]],\n",
       "\n",
       "         [[ 1.0004e+00, -1.2369e+00, -1.4535e+00,  ..., -6.1624e-01,\n",
       "            2.0532e+00,  7.6177e-01],\n",
       "          [-1.8414e+00, -1.1166e+00, -1.2063e+00,  ..., -1.4256e+00,\n",
       "            1.5698e+00,  1.1066e+00],\n",
       "          [-1.5353e-01, -4.5524e-01, -2.6076e+00,  ...,  1.9065e-01,\n",
       "           -2.7645e-01,  1.4270e+00],\n",
       "          ...,\n",
       "          [-1.8503e-01, -7.7978e-01, -3.2802e+00,  ..., -6.7907e-01,\n",
       "           -1.9723e+00,  3.2419e-01],\n",
       "          [ 4.2692e-01,  2.6884e-01, -5.9035e-01,  ...,  1.1083e+00,\n",
       "           -3.0354e-01,  3.2911e-01],\n",
       "          [ 1.3920e-01, -5.2097e-01,  1.2536e-01,  ...,  4.0581e-01,\n",
       "            7.9267e-01, -1.5163e-01]],\n",
       "\n",
       "         [[-6.7452e-01,  2.4451e-01, -1.2904e+00,  ...,  7.6509e-01,\n",
       "            2.4943e-01, -3.2176e-01],\n",
       "          [-1.8919e+00, -1.0580e+00,  8.2055e-01,  ...,  1.2977e+00,\n",
       "            3.4857e-02, -4.6494e-01],\n",
       "          [-1.4699e+00, -7.2937e-01, -7.7616e-01,  ...,  1.1342e+00,\n",
       "           -2.3719e+00, -1.9995e+00],\n",
       "          ...,\n",
       "          [-2.3746e+00, -2.2944e-01, -5.2514e-01,  ...,  1.1216e+00,\n",
       "            7.2139e-01,  4.0739e-01],\n",
       "          [-1.0875e+00, -1.3007e+00,  7.0438e-01,  ...,  4.9495e-02,\n",
       "           -5.0756e-01, -1.5508e+00],\n",
       "          [ 1.4159e-02,  3.7315e-01,  2.0663e-01,  ..., -7.9365e-02,\n",
       "           -6.8223e-01,  2.3441e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.6394e+00,  1.8518e-01,  5.2122e-01,  ..., -3.1063e-01,\n",
       "            4.5675e-01, -3.8960e-01],\n",
       "          [ 4.9246e-01,  9.0199e-01, -1.3637e+00,  ..., -9.5137e-01,\n",
       "           -3.5305e-01,  4.6194e-01],\n",
       "          [ 8.4342e-01,  2.1599e+00, -7.7921e-02,  ..., -2.0719e+00,\n",
       "           -1.2716e+00,  6.7825e-01],\n",
       "          ...,\n",
       "          [-9.7901e-01, -8.7847e-01,  7.8385e-01,  ...,  8.6403e-01,\n",
       "            6.7169e-01,  7.2932e-01],\n",
       "          [-5.1011e-01,  3.9475e-01,  9.2400e-01,  ...,  2.7086e-01,\n",
       "           -1.0948e+00,  9.0244e-01],\n",
       "          [ 5.3051e-03,  1.7702e-01,  3.3274e-01,  ..., -5.9739e-02,\n",
       "           -2.1845e-01, -5.2515e-01]],\n",
       "\n",
       "         [[ 1.4676e+00,  3.8106e-01, -4.7375e-01,  ..., -3.1602e-01,\n",
       "           -8.4914e-04,  1.8050e-01],\n",
       "          [-1.1618e+00, -2.7953e+00, -6.5261e-01,  ...,  3.3138e+00,\n",
       "            2.3612e+00, -1.0336e+00],\n",
       "          [-1.6697e+00, -1.4254e+00, -1.7867e+00,  ...,  1.3097e+00,\n",
       "            1.8563e+00, -1.6267e+00],\n",
       "          ...,\n",
       "          [-1.6595e+00,  5.1120e-01, -1.1573e+00,  ...,  4.1723e-01,\n",
       "            2.3718e+00,  1.3718e+00],\n",
       "          [-4.5326e-01, -4.2420e-01, -2.9853e-01,  ...,  1.2939e+00,\n",
       "            6.4114e-01,  1.4086e+00],\n",
       "          [ 2.3854e-01, -9.4748e-01, -1.2735e-01,  ...,  2.3971e-01,\n",
       "            3.1276e-02,  1.1632e+00]],\n",
       "\n",
       "         [[ 2.3543e+00, -1.1697e+00, -5.7613e-01,  ..., -4.9940e-01,\n",
       "            4.5801e-01,  8.3247e-01],\n",
       "          [ 6.0415e-01, -6.2720e-01,  2.0944e+00,  ...,  3.0404e+00,\n",
       "           -8.6737e-01, -5.9014e-01],\n",
       "          [-1.3971e+00, -5.2201e-01,  5.9768e-02,  ...,  2.0579e+00,\n",
       "            1.8793e+00,  1.5290e+00],\n",
       "          ...,\n",
       "          [ 1.2776e+00,  2.5898e-01, -9.3220e-01,  ...,  9.8087e-02,\n",
       "            1.2746e+00,  4.3148e-01],\n",
       "          [ 9.4816e-01,  4.5060e-01, -5.4849e-01,  ..., -2.4640e-01,\n",
       "            9.0641e-01, -1.5281e-01],\n",
       "          [ 4.2981e-01,  2.9019e-01, -3.5125e-01,  ..., -7.0732e-01,\n",
       "            6.0698e-01,  1.6006e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.7594e-01,  4.6932e-02,  7.0963e-01,  ..., -1.1336e+00,\n",
       "           -1.5239e+00, -3.2928e+00],\n",
       "          [ 2.0257e-01, -1.6692e+00,  1.7052e+00,  ...,  3.8870e-01,\n",
       "           -1.3552e+00, -1.9651e+00],\n",
       "          [ 1.3120e+00,  2.5840e-01,  1.0032e+00,  ...,  1.2524e+00,\n",
       "           -3.9550e+00, -1.2665e+00],\n",
       "          ...,\n",
       "          [ 2.5654e+00,  2.4158e-01,  1.7342e+00,  ..., -9.2253e-01,\n",
       "           -2.4702e+00, -3.5663e-01],\n",
       "          [ 1.0793e+00,  2.2171e-01,  2.8540e-02,  ...,  2.7037e-01,\n",
       "            5.9428e-01, -1.2530e+00],\n",
       "          [ 3.0909e-01, -2.8407e-02, -2.4523e-01,  ...,  1.2377e+00,\n",
       "           -8.4127e-01, -5.3714e-01]],\n",
       "\n",
       "         [[ 2.1847e-02, -2.0348e+00,  7.7939e-01,  ..., -6.0232e-01,\n",
       "            1.4656e+00, -2.9360e-01],\n",
       "          [-3.1498e-01,  3.5737e-01, -1.8155e+00,  ..., -3.7529e-01,\n",
       "           -9.8198e-01,  3.7630e-01],\n",
       "          [-1.7445e-02, -5.8202e-01, -8.7730e-02,  ..., -1.6204e+00,\n",
       "           -9.5201e-01,  6.4455e-02],\n",
       "          ...,\n",
       "          [-2.8139e+00,  3.2578e-01, -6.0799e-01,  ...,  9.1150e-01,\n",
       "           -9.1137e-01,  2.0762e+00],\n",
       "          [-1.1406e-01, -1.0860e+00, -5.5261e-01,  ...,  7.1286e-02,\n",
       "            3.6720e-01,  1.5755e+00],\n",
       "          [ 1.4775e-01, -5.7866e-01,  6.1598e-02,  ...,  3.9416e-01,\n",
       "            7.8993e-01, -1.2244e-01]],\n",
       "\n",
       "         [[-1.8429e+00,  9.8987e-02,  1.4858e+00,  ..., -7.6077e-01,\n",
       "            5.6919e-01,  8.0882e-01],\n",
       "          [-8.9898e-01, -6.5104e-01, -1.4980e+00,  ...,  4.7068e-01,\n",
       "           -2.0147e+00, -1.6881e+00],\n",
       "          [-6.3781e-01, -1.5039e+00,  2.3245e+00,  ...,  7.9916e-01,\n",
       "           -1.9842e+00, -9.1488e-01],\n",
       "          ...,\n",
       "          [-1.0800e+00,  3.3917e-01,  2.7692e-02,  ...,  1.8424e+00,\n",
       "           -2.7900e-01,  6.9194e-01],\n",
       "          [-5.8747e-01, -2.1277e-01, -8.9565e-01,  ...,  1.4535e+00,\n",
       "           -3.5486e-01, -2.4431e-01],\n",
       "          [-7.3580e-03,  4.2430e-01,  1.8634e-01,  ..., -2.4569e-02,\n",
       "           -7.1955e-01,  2.3433e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.9027e+00, -1.1780e+00, -1.5464e+00,  ..., -5.4422e-01,\n",
       "           -4.3158e-01,  5.9128e-01],\n",
       "          [-7.3711e-01,  3.3978e-01, -1.1894e+00,  ..., -1.0545e+00,\n",
       "           -2.7587e+00,  1.6910e+00],\n",
       "          [-2.7010e-01,  4.4338e-01, -1.3199e+00,  ...,  1.0964e-01,\n",
       "           -1.3790e+00, -9.5369e-01],\n",
       "          ...,\n",
       "          [-1.5703e+00,  2.4047e-01, -1.1879e+00,  ..., -7.6170e-01,\n",
       "           -2.8923e+00, -1.0154e+00],\n",
       "          [-3.5685e-01,  9.2990e-01,  1.9611e-01,  ...,  3.2104e-01,\n",
       "           -1.8896e-01,  3.1868e-01],\n",
       "          [ 4.3837e-02,  1.8806e-01,  3.5428e-01,  ..., -6.2998e-02,\n",
       "           -2.2245e-01, -5.3558e-01]],\n",
       "\n",
       "         [[ 1.3020e+00,  2.4117e-01,  1.4926e+00,  ..., -2.8298e-01,\n",
       "           -9.8047e-01,  2.1183e-01],\n",
       "          [-3.0637e-01, -2.2697e+00,  2.1488e-03,  ...,  2.1875e+00,\n",
       "           -4.7616e-01,  5.5323e-01],\n",
       "          [-2.7198e+00, -9.4302e-01,  8.5966e-01,  ...,  2.7828e+00,\n",
       "            1.7945e+00,  2.4525e+00],\n",
       "          ...,\n",
       "          [-1.1074e+00, -1.5962e+00,  1.3464e-01,  ...,  2.2650e+00,\n",
       "            1.0436e+00,  6.7609e-01],\n",
       "          [-1.4027e+00, -4.9645e-01, -1.4404e+00,  ...,  9.2851e-01,\n",
       "            1.5958e+00,  1.1062e+00],\n",
       "          [ 2.0113e-01, -9.5454e-01, -1.1345e-01,  ...,  2.5207e-01,\n",
       "            8.6281e-03,  1.1896e+00]],\n",
       "\n",
       "         [[ 3.8094e-01, -9.8767e-01, -2.3484e+00,  ...,  1.1204e+00,\n",
       "            1.5208e+00, -1.3631e+00],\n",
       "          [ 2.6237e-01,  2.6176e-01, -3.7777e-01,  ...,  6.0121e-01,\n",
       "            2.1216e+00,  3.1849e-01],\n",
       "          [ 1.2264e-01, -1.6869e+00, -8.2065e-01,  ...,  1.7734e-01,\n",
       "            2.3168e+00, -9.4054e-02],\n",
       "          ...,\n",
       "          [-2.4729e-01,  2.8854e-01, -5.8045e-01,  ...,  3.1551e-01,\n",
       "           -1.4320e+00,  2.9104e-01],\n",
       "          [ 3.6977e-01, -2.1946e-01, -6.4355e-01,  ..., -1.3862e-01,\n",
       "            1.5923e+00, -1.9332e-01],\n",
       "          [ 4.3979e-01,  2.7464e-01, -3.8203e-01,  ..., -7.3983e-01,\n",
       "            6.5693e-01,  2.1223e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.2280e+00, -8.2414e-01,  1.1393e+00,  ...,  1.5200e+00,\n",
       "            2.4128e-01, -1.3591e+00],\n",
       "          [-2.7785e-01, -2.0768e+00,  1.3798e+00,  ..., -6.0140e-01,\n",
       "           -1.4403e+00, -1.5620e+00],\n",
       "          [-3.3824e+00,  1.6143e-01,  2.1305e+00,  ..., -1.0832e+00,\n",
       "            8.7178e-01, -5.8205e-01],\n",
       "          ...,\n",
       "          [ 6.2259e-01, -7.4973e-01,  8.1083e-01,  ...,  9.5323e-01,\n",
       "           -2.0682e+00, -3.6466e-01],\n",
       "          [-9.9020e-01, -4.7138e-01,  2.2918e-01,  ..., -1.0654e-01,\n",
       "           -1.1863e+00, -4.0885e-01],\n",
       "          [ 2.8050e-01, -2.2476e-02, -2.2661e-01,  ...,  1.2276e+00,\n",
       "           -8.8584e-01, -4.9024e-01]],\n",
       "\n",
       "         [[-1.6033e+00, -1.8909e-01, -1.0743e+00,  ..., -1.2653e-01,\n",
       "           -8.3144e-01, -1.2088e+00],\n",
       "          [ 4.1280e-01, -1.2644e-02, -1.5126e+00,  ..., -1.9388e+00,\n",
       "           -3.0079e-01,  1.8618e+00],\n",
       "          [ 2.5168e-02,  1.4188e+00,  2.1981e-01,  ..., -1.7483e-01,\n",
       "           -2.9612e-01, -2.2898e-01],\n",
       "          ...,\n",
       "          [ 1.2474e+00, -9.8258e-01, -1.6188e+00,  ..., -1.0096e+00,\n",
       "            8.1785e-01,  2.2308e+00],\n",
       "          [-5.3781e-01, -1.2658e+00, -8.4016e-01,  ...,  6.9985e-01,\n",
       "            1.0284e+00,  1.8909e-01],\n",
       "          [ 1.3874e-01, -5.8009e-01,  1.3843e-01,  ...,  4.1072e-01,\n",
       "            8.2594e-01, -1.1441e-01]],\n",
       "\n",
       "         [[-1.1054e+00,  1.8610e+00, -2.6322e-01,  ...,  1.0125e+00,\n",
       "           -9.4373e-01, -4.6625e-01],\n",
       "          [-1.6913e+00, -2.9662e+00,  2.0950e+00,  ...,  1.4132e+00,\n",
       "           -1.8183e+00,  2.0273e-01],\n",
       "          [-1.5206e+00,  7.1946e-01, -4.4659e-01,  ..., -2.0552e-01,\n",
       "           -1.8080e+00,  2.0800e+00],\n",
       "          ...,\n",
       "          [ 3.0954e-01,  3.4728e-01,  1.8809e+00,  ..., -1.1127e+00,\n",
       "           -1.3230e+00,  2.0798e-01],\n",
       "          [ 2.7138e-02,  2.4607e-01,  4.4594e-01,  ...,  3.6973e-04,\n",
       "            1.5654e-01, -6.6298e-01],\n",
       "          [-4.4441e-03,  4.0478e-01,  2.2549e-01,  ..., -3.0574e-02,\n",
       "           -6.7538e-01,  2.0440e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-6.4532e-01,  1.2565e-01, -3.4848e-01,  ...,  8.3598e-01,\n",
       "            1.4141e+00,  9.4007e-01],\n",
       "          [ 9.1005e-02, -5.9622e-01, -5.6969e-01,  ...,  4.6402e-01,\n",
       "            1.7426e-01,  2.5544e-01],\n",
       "          [-2.2903e-01, -2.5099e-01, -4.8405e-01,  ...,  7.5017e-01,\n",
       "           -3.9636e-01,  1.5679e-01],\n",
       "          ...,\n",
       "          [ 5.5496e-01, -3.0282e-01,  3.5972e-02,  ...,  3.3375e-01,\n",
       "            9.3176e-02,  3.4795e-01],\n",
       "          [ 1.0382e+00, -4.6478e-01, -7.0726e-01,  ..., -4.0385e-02,\n",
       "            5.5952e-01,  9.4941e-01],\n",
       "          [-8.6116e-01, -3.3357e-01, -8.0866e-01,  ...,  9.3549e-01,\n",
       "            1.0176e-01, -8.3726e-01]],\n",
       "\n",
       "         [[-6.1700e-01,  1.6919e-01,  8.5537e-01,  ...,  2.7609e-01,\n",
       "            9.1863e-01,  8.8257e-01],\n",
       "          [-4.3740e-01,  2.6108e-01, -9.1502e-01,  ...,  7.1496e-01,\n",
       "            7.0282e-01,  2.5017e-01],\n",
       "          [ 8.9235e-02,  3.1912e-01, -3.0358e-01,  ...,  5.0172e-01,\n",
       "            5.6920e-01,  4.2265e-01],\n",
       "          ...,\n",
       "          [-4.9633e-01,  6.5425e-01, -4.3804e-01,  ..., -5.5873e-01,\n",
       "           -4.2032e-01,  1.0496e+00],\n",
       "          [-6.6853e-01,  3.4918e-01, -9.6809e-01,  ..., -2.8175e-01,\n",
       "           -4.9466e-01,  4.8158e-01],\n",
       "          [ 7.0918e-01, -1.4118e-01, -4.3066e-01,  ...,  1.2678e+00,\n",
       "           -1.2817e-01,  2.4008e+00]],\n",
       "\n",
       "         [[-3.7963e-01, -1.6276e+00,  3.5491e-01,  ...,  9.6683e-01,\n",
       "           -3.2882e-01, -2.5011e-01],\n",
       "          [-9.7870e-01, -8.2441e-01,  9.7145e-01,  ...,  1.1722e-01,\n",
       "            5.1670e-01, -1.0095e+00],\n",
       "          [-5.2288e-01, -7.7682e-02,  1.0753e+00,  ..., -4.9061e-01,\n",
       "            4.9566e-01, -8.9173e-01],\n",
       "          ...,\n",
       "          [-1.0553e+00, -2.3988e-01,  1.0593e-01,  ...,  1.0642e+00,\n",
       "           -1.1376e-01,  1.4212e-01],\n",
       "          [-8.2974e-01, -5.8765e-01,  6.7888e-01,  ...,  1.5758e+00,\n",
       "           -5.8612e-01, -1.2438e-02],\n",
       "          [-4.2354e-01, -8.7837e-01,  1.3130e+00,  ...,  9.8310e-01,\n",
       "           -1.0012e+00,  2.3824e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 9.6707e-01, -5.7835e-01,  1.1234e+00,  ...,  6.1529e-01,\n",
       "            1.0579e+00, -4.3005e-01],\n",
       "          [ 2.1471e-01, -6.1530e-01,  4.1110e-01,  ...,  8.1865e-01,\n",
       "            9.5823e-01,  9.2309e-02],\n",
       "          [ 4.4636e-01, -1.0164e+00,  1.5626e-01,  ...,  1.0267e+00,\n",
       "            1.0788e+00,  6.3671e-02],\n",
       "          ...,\n",
       "          [ 4.4506e-01, -3.1192e-01,  1.8808e-01,  ...,  1.2192e+00,\n",
       "            4.3517e-01, -9.0311e-01],\n",
       "          [ 2.4876e-01, -4.7374e-01,  9.3455e-02,  ...,  4.4534e-01,\n",
       "           -1.8199e-01, -4.1731e-01],\n",
       "          [ 1.0587e-01,  1.0767e+00, -3.2117e-01,  ..., -1.7245e-01,\n",
       "           -2.6706e-01,  1.5456e+00]],\n",
       "\n",
       "         [[ 9.2732e-02, -8.8136e-01,  1.7598e+00,  ...,  1.2900e+00,\n",
       "            8.4476e-01,  1.0456e+00],\n",
       "          [-3.2474e-01,  2.5068e-01, -7.7383e-01,  ..., -6.7234e-01,\n",
       "           -5.3407e-01,  1.7614e+00],\n",
       "          [ 2.4621e-01, -2.9330e-01, -8.7214e-01,  ...,  9.9390e-02,\n",
       "           -7.9296e-02,  3.8336e-01],\n",
       "          ...,\n",
       "          [-1.3102e+00,  7.1531e-01, -1.0723e+00,  ..., -5.4729e-01,\n",
       "           -6.7172e-01,  1.0549e+00],\n",
       "          [-7.5773e-01,  1.1460e+00, -1.1998e+00,  ..., -2.7833e-01,\n",
       "           -8.7304e-01,  1.2154e+00],\n",
       "          [-5.9930e-01,  1.5969e+00, -1.4549e+00,  ...,  6.6832e-01,\n",
       "           -1.5033e+00, -7.4270e-01]],\n",
       "\n",
       "         [[-2.0260e-01,  6.4208e-01, -3.1249e-02,  ..., -3.0778e-01,\n",
       "           -8.4674e-01, -7.0874e-01],\n",
       "          [ 3.3639e-01, -1.4233e-01, -1.2580e-01,  ..., -4.1002e-01,\n",
       "            6.4454e-01, -2.8666e-01],\n",
       "          [ 3.3086e-01, -4.9568e-01,  7.2109e-01,  ..., -3.4298e-01,\n",
       "            5.4471e-01, -5.4634e-01],\n",
       "          ...,\n",
       "          [ 9.9506e-01, -1.0148e-01,  1.0932e-01,  ..., -7.2174e-01,\n",
       "            2.2213e-01, -8.0974e-01],\n",
       "          [ 5.6054e-01,  2.9071e-01,  6.6759e-01,  ..., -4.8374e-01,\n",
       "           -3.7016e-01, -7.9906e-01],\n",
       "          [ 1.2098e+00, -6.7077e-01, -9.3181e-01,  ..., -1.1607e+00,\n",
       "            1.6049e+00, -6.1710e-01]]],\n",
       "\n",
       "\n",
       "        [[[-6.4446e-01,  1.1206e-01, -3.6608e-01,  ...,  8.5483e-01,\n",
       "            1.4181e+00,  9.5373e-01],\n",
       "          [ 9.4689e-01,  4.8202e-03, -1.4840e+00,  ...,  7.2230e-01,\n",
       "            1.9540e+00,  1.9908e+00],\n",
       "          [ 3.8595e-01, -1.4359e+00, -1.2732e+00,  ...,  1.0794e+00,\n",
       "            6.0180e-01,  5.5094e-01],\n",
       "          ...,\n",
       "          [-7.0772e-02,  2.2640e-01, -8.4898e-01,  ...,  8.1467e-01,\n",
       "            8.7101e-01,  1.0144e+00],\n",
       "          [-1.7591e-01, -3.5221e-02, -7.7914e-01,  ...,  1.1568e+00,\n",
       "            5.4986e-01,  8.2372e-01],\n",
       "          [-1.5783e-01, -1.3233e-01, -2.0439e-01,  ...,  1.2315e+00,\n",
       "            5.9073e-01,  1.1962e+00]],\n",
       "\n",
       "         [[-6.2048e-01,  1.7466e-01,  8.4459e-01,  ...,  3.0080e-01,\n",
       "            9.1947e-01,  8.9195e-01],\n",
       "          [-8.7777e-01,  4.6724e-01, -7.2848e-01,  ...,  5.6768e-01,\n",
       "           -1.0287e+00,  9.5174e-01],\n",
       "          [-6.7991e-01,  5.1922e-01, -4.6603e-01,  ...,  4.1863e-01,\n",
       "            2.0874e-01,  4.9544e-01],\n",
       "          ...,\n",
       "          [-5.8414e-01, -4.4329e-01, -5.4566e-01,  ...,  6.3139e-01,\n",
       "           -1.1526e+00,  1.4640e+00],\n",
       "          [-9.4976e-01,  5.1556e-01, -2.3569e-01,  ...,  4.7326e-01,\n",
       "            1.9920e-01,  8.7700e-01],\n",
       "          [ 1.5696e-01, -9.0682e-01, -1.2500e+00,  ...,  5.9036e-01,\n",
       "           -2.3001e-01,  1.3565e+00]],\n",
       "\n",
       "         [[-3.4788e-01, -1.5948e+00,  3.7582e-01,  ...,  9.8984e-01,\n",
       "           -2.9633e-01, -3.0765e-01],\n",
       "          [-3.8312e-01, -1.6242e+00,  9.5136e-01,  ...,  9.7191e-01,\n",
       "            1.0216e-01, -6.3176e-01],\n",
       "          [-7.2823e-01, -6.9365e-01,  8.0342e-01,  ...,  8.3544e-01,\n",
       "           -2.1747e-01,  3.1056e-01],\n",
       "          ...,\n",
       "          [-9.7416e-01, -6.1229e-01,  6.5070e-01,  ...,  6.9179e-01,\n",
       "           -6.9047e-01, -2.3431e-01],\n",
       "          [-1.0304e+00, -8.2766e-01,  1.0479e+00,  ...,  1.7724e+00,\n",
       "           -3.4719e-01, -7.2930e-01],\n",
       "          [-9.4388e-02, -3.1525e-01,  1.7036e+00,  ...,  1.5759e+00,\n",
       "           -4.6596e-01,  1.4946e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 9.4660e-01, -5.6891e-01,  1.1329e+00,  ...,  5.8068e-01,\n",
       "            1.0587e+00, -4.4622e-01],\n",
       "          [-1.3911e+00, -1.8500e-01,  9.6108e-01,  ...,  5.3238e-02,\n",
       "           -1.6271e-01, -1.8392e-02],\n",
       "          [-8.7396e-03, -8.5423e-01, -7.1990e-01,  ...,  4.5060e-01,\n",
       "            2.6260e-01, -5.2762e-01],\n",
       "          ...,\n",
       "          [-7.6798e-01, -2.2006e-01,  6.8987e-01,  ..., -4.7705e-01,\n",
       "           -3.5049e-01, -5.0428e-01],\n",
       "          [ 4.4836e-01, -3.2761e-01,  6.5500e-01,  ...,  5.8002e-01,\n",
       "           -2.0738e-01, -5.8809e-01],\n",
       "          [-3.0372e-02,  5.6393e-03, -1.0520e+00,  ...,  1.9215e-01,\n",
       "           -3.7423e-01, -2.8831e-01]],\n",
       "\n",
       "         [[ 3.0952e-02, -8.8506e-01,  1.7551e+00,  ...,  1.2913e+00,\n",
       "            8.4590e-01,  1.0302e+00],\n",
       "          [ 4.2478e-01,  1.3079e+00, -8.0032e-01,  ..., -3.7933e-01,\n",
       "           -1.0647e+00,  1.5340e+00],\n",
       "          [-7.8567e-01,  7.9373e-01, -1.2336e+00,  ..., -1.0317e-01,\n",
       "           -4.5355e-01,  6.2032e-01],\n",
       "          ...,\n",
       "          [-1.4162e+00,  1.6877e-01, -8.8566e-01,  ...,  2.2957e-02,\n",
       "            1.6181e-01,  2.8519e-01],\n",
       "          [-3.5640e-01,  5.7463e-01, -3.0114e-01,  ...,  1.1517e-01,\n",
       "           -3.2206e-02,  1.4079e+00],\n",
       "          [-1.2497e+00,  5.3125e-01, -7.6714e-01,  ...,  6.6571e-01,\n",
       "           -1.1592e+00,  1.1695e+00]],\n",
       "\n",
       "         [[-1.7740e-01,  6.8144e-01, -5.1662e-02,  ..., -3.1682e-01,\n",
       "           -8.6909e-01, -7.3014e-01],\n",
       "          [ 7.6746e-02, -8.0370e-01,  6.0188e-01,  ..., -1.7623e+00,\n",
       "           -4.2077e-01,  8.5548e-02],\n",
       "          [ 5.2916e-01,  5.6401e-02,  6.3413e-02,  ..., -5.7233e-01,\n",
       "           -3.1745e-01,  8.6647e-02],\n",
       "          ...,\n",
       "          [ 1.4860e+00,  1.1650e-01,  1.0185e+00,  ..., -9.2415e-01,\n",
       "            2.3956e-01, -1.5218e+00],\n",
       "          [-2.7294e-01,  7.1223e-01,  1.7636e-01,  ...,  2.7925e-02,\n",
       "           -6.3133e-01, -5.6057e-01],\n",
       "          [ 5.4836e-01,  7.6210e-02, -6.4434e-01,  ..., -3.2700e-01,\n",
       "            7.7983e-01, -5.9397e-01]]],\n",
       "\n",
       "\n",
       "        [[[-6.7491e-01,  1.4415e-01, -3.5766e-01,  ...,  8.4898e-01,\n",
       "            1.3889e+00,  9.3228e-01],\n",
       "          [-1.6384e-01, -2.9954e-01, -8.4976e-01,  ...,  4.1931e-01,\n",
       "            1.1728e+00,  7.7348e-01],\n",
       "          [-1.0469e+00,  6.2592e-02, -8.5627e-01,  ...,  8.2124e-01,\n",
       "            8.0133e-01,  3.8303e-01],\n",
       "          ...,\n",
       "          [ 6.1501e-01, -5.0392e-01, -5.8128e-01,  ..., -4.3737e-01,\n",
       "            7.2204e-01,  3.9291e-01],\n",
       "          [-1.8203e-01, -2.4660e-01, -9.2960e-01,  ...,  8.6842e-01,\n",
       "            7.0069e-01,  1.0674e+00],\n",
       "          [-6.3768e-01, -1.0039e-01, -2.7081e-01,  ...,  4.3529e-01,\n",
       "            7.1790e-01,  4.7713e-01]],\n",
       "\n",
       "         [[-6.4418e-01,  1.4740e-01,  8.5399e-01,  ...,  2.8677e-01,\n",
       "            9.2422e-01,  8.6072e-01],\n",
       "          [-1.0330e+00, -5.3013e-01, -1.0106e+00,  ...,  3.8487e-02,\n",
       "           -7.9309e-02,  9.0521e-01],\n",
       "          [-4.9610e-01, -4.4094e-01, -4.8467e-01,  ...,  5.5574e-02,\n",
       "            1.5147e-01,  3.6519e-01],\n",
       "          ...,\n",
       "          [-7.3979e-01, -2.4725e-02, -1.4503e+00,  ...,  4.0130e-01,\n",
       "           -1.8796e-01,  3.5017e-01],\n",
       "          [-4.6203e-01, -2.5573e-01, -1.5001e+00,  ..., -4.1504e-02,\n",
       "           -8.5692e-01,  3.3567e-01],\n",
       "          [-7.6580e-02, -8.8293e-01, -1.6350e+00,  ...,  1.3857e+00,\n",
       "            3.0033e-02,  1.7342e+00]],\n",
       "\n",
       "         [[-3.3542e-01, -1.6015e+00,  3.6186e-01,  ...,  9.8671e-01,\n",
       "           -2.8581e-01, -2.8505e-01],\n",
       "          [ 1.0534e-01, -1.3093e+00,  9.0958e-01,  ...,  1.5104e+00,\n",
       "            2.0763e-01,  1.3642e-01],\n",
       "          [ 1.3115e-01, -7.5042e-01,  8.1464e-01,  ...,  1.1158e+00,\n",
       "           -2.6212e-01,  6.0642e-01],\n",
       "          ...,\n",
       "          [ 4.2095e-01, -1.1287e+00,  1.1024e+00,  ...,  1.2480e+00,\n",
       "           -5.4324e-01,  5.0282e-01],\n",
       "          [ 1.0477e-01, -6.3146e-01,  1.1794e+00,  ...,  1.0619e+00,\n",
       "           -1.0634e-01, -4.8722e-01],\n",
       "          [-2.3774e-01, -1.0590e+00,  1.3630e+00,  ...,  1.2365e+00,\n",
       "           -1.1221e+00,  7.2978e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 9.5584e-01, -5.7519e-01,  1.1304e+00,  ...,  6.0645e-01,\n",
       "            1.0405e+00, -4.4947e-01],\n",
       "          [ 6.3291e-01, -5.2062e-01, -1.7410e-01,  ...,  2.0792e-03,\n",
       "            2.6252e-01, -3.9011e-01],\n",
       "          [ 1.6262e-01, -3.3571e-01,  3.2479e-02,  ...,  1.4267e-01,\n",
       "           -5.9476e-01, -1.1105e+00],\n",
       "          ...,\n",
       "          [ 9.8077e-01, -3.2888e-01,  5.0123e-01,  ..., -1.6148e-02,\n",
       "           -1.8738e-01, -6.2248e-01],\n",
       "          [ 1.0285e-01,  2.3793e-02, -2.7548e-02,  ..., -4.5619e-02,\n",
       "            2.9279e-01, -4.7788e-01],\n",
       "          [ 1.5350e+00,  2.3872e-01, -8.8196e-01,  ..., -1.3139e-01,\n",
       "           -9.2000e-01,  6.5956e-01]],\n",
       "\n",
       "         [[ 6.7345e-02, -9.0718e-01,  1.7919e+00,  ...,  1.2882e+00,\n",
       "            8.3090e-01,  9.9516e-01],\n",
       "          [ 4.9807e-01,  6.3233e-01, -5.4423e-01,  ...,  3.4243e-01,\n",
       "            3.9307e-01, -2.9838e-01],\n",
       "          [ 6.2356e-01, -4.3979e-02,  2.3030e-01,  ...,  1.2603e+00,\n",
       "           -1.7627e-01, -3.4621e-01],\n",
       "          ...,\n",
       "          [-7.7813e-01,  4.6405e-02, -1.1059e+00,  ...,  7.4448e-02,\n",
       "           -4.7867e-01,  3.2548e-02],\n",
       "          [-7.1498e-01, -5.4633e-01, -9.8048e-01,  ...,  5.8960e-01,\n",
       "           -2.6851e-01,  2.2587e-01],\n",
       "          [-1.8840e+00,  1.7822e+00, -1.4574e+00,  ..., -7.3492e-01,\n",
       "           -7.6305e-01,  9.3820e-01]],\n",
       "\n",
       "         [[-1.8959e-01,  6.6688e-01, -2.7591e-02,  ..., -2.9949e-01,\n",
       "           -8.5770e-01, -7.2926e-01],\n",
       "          [ 1.7708e-01, -1.0099e-01,  6.7663e-01,  ..., -8.1937e-01,\n",
       "            4.0675e-01, -5.5706e-01],\n",
       "          [-8.0559e-01, -1.4063e-02, -1.1806e+00,  ...,  1.6627e-01,\n",
       "           -2.5221e-02, -1.0331e+00],\n",
       "          ...,\n",
       "          [ 6.0553e-01,  2.4881e-01,  6.8113e-01,  ..., -5.0662e-01,\n",
       "            8.2221e-02, -3.3691e-01],\n",
       "          [ 3.4765e-01, -1.2909e+00,  5.6408e-03,  ..., -1.0520e-01,\n",
       "            8.9379e-01, -1.8647e-01],\n",
       "          [ 1.6081e+00, -3.6272e-01, -3.1999e-01,  ..., -7.8213e-01,\n",
       "            1.2613e+00, -2.7594e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-3.8473e-01, -8.0442e-02,  2.5350e-01,  ..., -1.6875e-01,\n",
       "           -3.9282e-01,  2.2661e-01],\n",
       "          [ 4.4996e-01, -5.2405e-01,  2.9155e-01,  ...,  8.5549e-02,\n",
       "            1.4100e-02, -4.0176e-01],\n",
       "          [ 4.1911e-01, -5.1707e-01, -2.4712e-01,  ...,  6.7625e-01,\n",
       "           -2.4651e-02, -1.6581e-01],\n",
       "          ...,\n",
       "          [-1.2768e-01,  2.5614e-02, -1.8311e-02,  ...,  8.6133e-01,\n",
       "            6.8822e-01,  6.6512e-02],\n",
       "          [-1.4794e-02,  3.1781e-01, -2.3705e-01,  ...,  2.1798e-01,\n",
       "           -1.8075e-01, -1.1050e-01],\n",
       "          [-2.5591e-01,  1.5606e+00,  6.5310e-01,  ..., -8.2905e-02,\n",
       "           -4.0437e-01, -4.7523e-01]],\n",
       "\n",
       "         [[ 1.2992e-01,  7.5396e-01,  3.5269e-01,  ...,  2.6549e-01,\n",
       "           -7.6103e-03,  5.0459e-01],\n",
       "          [-1.1923e-01, -1.7441e-02, -7.5107e-02,  ..., -2.9604e-01,\n",
       "           -5.6270e-01, -6.3833e-02],\n",
       "          [-4.9901e-01, -8.9582e-02, -5.5655e-01,  ..., -2.4703e-02,\n",
       "           -3.6618e-01, -1.6988e-01],\n",
       "          ...,\n",
       "          [ 3.0201e-01, -7.1945e-01, -1.0688e-01,  ..., -4.0877e-01,\n",
       "            5.0405e-01,  1.8876e-01],\n",
       "          [ 1.8801e-01, -2.2696e-01, -8.5216e-02,  ..., -4.3135e-01,\n",
       "            1.3162e-01, -4.0985e-02],\n",
       "          [-3.7618e-01,  3.1486e-01, -5.6942e-01,  ..., -4.3385e-01,\n",
       "           -1.1025e+00, -5.5280e-01]],\n",
       "\n",
       "         [[-1.1017e+00, -4.5148e-01, -8.2765e-01,  ..., -3.2913e-01,\n",
       "           -3.4708e-01, -2.0972e-01],\n",
       "          [-5.7512e-01, -2.7297e-01, -3.8519e-01,  ..., -6.3137e-01,\n",
       "            4.0667e-01,  4.9277e-01],\n",
       "          [-6.1987e-01, -4.6815e-01,  3.3199e-02,  ..., -4.4856e-01,\n",
       "            3.4972e-01,  5.8794e-01],\n",
       "          ...,\n",
       "          [-3.7160e-01, -6.9512e-01, -3.5287e-01,  ..., -1.6835e-01,\n",
       "            4.1750e-01,  7.0798e-02],\n",
       "          [ 5.0184e-01, -7.5437e-01, -5.2158e-01,  ...,  3.3748e-01,\n",
       "           -5.3436e-01,  2.6481e-01],\n",
       "          [-1.5833e+00, -8.6473e-01, -1.1233e+00,  ..., -3.7456e-01,\n",
       "            2.6460e-01,  1.0767e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.1095e-01,  8.9553e-01,  6.1992e-02,  ..., -4.7154e-01,\n",
       "            2.6863e-01, -2.8858e-01],\n",
       "          [-5.6605e-01, -3.6724e-02, -4.9370e-01,  ...,  2.1603e-01,\n",
       "           -4.1152e-01,  5.6708e-02],\n",
       "          [-1.3374e-01,  2.2700e-02, -1.2079e-01,  ...,  5.4669e-01,\n",
       "           -1.9678e-02, -2.0661e-01],\n",
       "          ...,\n",
       "          [-1.0925e-01,  1.9570e-01,  4.2646e-01,  ..., -6.2075e-01,\n",
       "           -5.5178e-01,  1.3287e-01],\n",
       "          [-5.0502e-01,  6.3828e-01, -3.5246e-01,  ..., -3.0522e-01,\n",
       "           -5.0051e-02,  3.2674e-01],\n",
       "          [ 9.7413e-01,  9.4546e-01, -5.9052e-01,  ...,  2.5318e-02,\n",
       "            2.1698e+00,  1.5707e-01]],\n",
       "\n",
       "         [[-4.6330e-01, -2.1819e+00, -8.5137e-01,  ..., -8.9747e-02,\n",
       "           -3.7621e-01, -2.0053e+00],\n",
       "          [-4.2787e-01,  2.2555e-01,  4.7290e-01,  ...,  2.1714e-01,\n",
       "            3.3386e-01,  7.9678e-02],\n",
       "          [-5.2589e-01, -1.6014e-01, -8.8063e-02,  ...,  4.0473e-01,\n",
       "            1.5030e-01,  1.1563e-01],\n",
       "          ...,\n",
       "          [ 2.4599e-01,  1.6882e-01,  3.8827e-01,  ..., -7.6973e-02,\n",
       "           -1.7004e-01, -4.6313e-03],\n",
       "          [ 2.6316e-01,  2.8132e-01, -2.9939e-01,  ...,  6.3059e-01,\n",
       "            2.9850e-01,  4.0876e-03],\n",
       "          [ 3.5846e-01, -8.6602e-01,  9.5159e-01,  ..., -4.3857e-01,\n",
       "            4.9680e-01, -9.3024e-01]],\n",
       "\n",
       "         [[-5.4027e-01, -4.0881e-01, -2.6557e-01,  ..., -9.9184e-01,\n",
       "            5.2985e-01, -5.2104e-01],\n",
       "          [ 9.1605e-01,  6.2366e-01, -4.7439e-01,  ...,  7.2774e-01,\n",
       "            4.1656e-01,  3.7127e-01],\n",
       "          [-6.1802e-01, -3.4765e-01, -3.3265e-01,  ...,  4.1035e-01,\n",
       "            3.4262e-02, -1.8105e-04],\n",
       "          ...,\n",
       "          [-1.1136e-01, -1.2864e-02, -1.0587e+00,  ..., -2.6361e-01,\n",
       "            5.7522e-01,  6.1039e-01],\n",
       "          [ 9.1656e-02, -8.8071e-01, -1.7276e-01,  ..., -1.0577e-01,\n",
       "           -3.7623e-01,  1.8881e-01],\n",
       "          [-5.0741e-01,  1.3183e+00, -3.1871e-01,  ..., -9.0109e-01,\n",
       "           -1.5673e-01,  1.2724e-01]]],\n",
       "\n",
       "\n",
       "        [[[-3.7700e-01, -4.7210e-02,  2.7021e-01,  ..., -1.7596e-01,\n",
       "           -3.9876e-01,  2.0669e-01],\n",
       "          [-9.4434e-01,  1.1396e+00,  1.2511e+00,  ...,  4.8076e-01,\n",
       "           -4.7667e-01, -3.5001e-01],\n",
       "          [ 2.9758e-01,  4.1508e-01,  4.4479e-01,  ...,  1.4212e-01,\n",
       "           -5.0463e-02,  2.6746e-01],\n",
       "          ...,\n",
       "          [-7.7776e-01,  1.2502e+00, -6.6100e-02,  ...,  1.6546e-01,\n",
       "            7.0697e-02,  1.2370e-01],\n",
       "          [-7.7889e-01,  3.2454e-01, -2.3255e-03,  ...,  3.4185e-01,\n",
       "            5.6222e-01,  3.4216e-01],\n",
       "          [ 4.1203e-01,  1.2835e+00,  2.2446e-01,  ...,  4.2623e-01,\n",
       "           -4.2853e-02,  6.1821e-01]],\n",
       "\n",
       "         [[ 1.1568e-01,  7.6606e-01,  4.0462e-01,  ...,  2.4925e-01,\n",
       "            1.2137e-02,  5.1048e-01],\n",
       "          [ 3.6204e-01,  3.6777e-01,  2.7767e-01,  ...,  4.1480e-01,\n",
       "            8.9939e-01,  1.2313e+00],\n",
       "          [-7.0837e-02,  8.9246e-02, -3.8567e-01,  ..., -1.4605e-01,\n",
       "            6.0949e-01,  3.9990e-01],\n",
       "          ...,\n",
       "          [-3.1743e-01, -1.6200e-01, -7.3801e-01,  ..., -4.7802e-01,\n",
       "           -2.9338e-02, -1.9705e-01],\n",
       "          [ 6.9565e-02,  1.6431e-01, -7.6505e-02,  ..., -2.0898e-01,\n",
       "           -1.5268e-01,  3.1641e-01],\n",
       "          [ 5.1106e-01,  1.6551e-03, -3.5943e-01,  ..., -1.1941e-01,\n",
       "            1.9484e-01,  6.1941e-01]],\n",
       "\n",
       "         [[-1.1158e+00, -4.5519e-01, -8.3062e-01,  ..., -3.4530e-01,\n",
       "           -3.5024e-01, -2.3029e-01],\n",
       "          [-1.0686e+00, -7.1558e-01, -8.0963e-01,  ..., -1.0976e+00,\n",
       "           -2.6558e-01,  6.2751e-01],\n",
       "          [-5.4305e-01, -7.2394e-01,  1.3242e-01,  ..., -2.5362e-01,\n",
       "           -3.5368e-01,  2.2427e-01],\n",
       "          ...,\n",
       "          [-8.7954e-01, -6.5177e-01, -3.7928e-01,  ...,  2.7572e-02,\n",
       "           -2.3263e-01, -2.2107e-01],\n",
       "          [-8.6735e-01, -7.1411e-01, -9.6456e-01,  ..., -7.9104e-01,\n",
       "           -5.6781e-02,  4.3093e-01],\n",
       "          [-1.8084e+00, -8.6109e-01, -4.9097e-01,  ..., -3.0329e-01,\n",
       "            2.5572e-01,  7.3617e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.3278e-01,  9.0467e-01,  5.8046e-02,  ..., -4.9722e-01,\n",
       "            2.9957e-01, -2.6818e-01],\n",
       "          [ 1.2008e-01,  3.3317e-01,  6.8419e-01,  ...,  5.6369e-01,\n",
       "            2.3227e-01, -1.6853e-02],\n",
       "          [-1.8114e-01, -1.8131e-01,  1.4698e-01,  ...,  5.2401e-01,\n",
       "            4.3829e-01, -1.2287e-02],\n",
       "          ...,\n",
       "          [ 2.6674e-01,  7.7291e-01, -7.2826e-02,  ...,  6.4343e-01,\n",
       "            2.3198e-01,  7.7015e-02],\n",
       "          [-1.7930e-01,  9.7932e-01,  5.0694e-02,  ..., -1.0369e+00,\n",
       "           -4.4488e-01, -1.0677e-01],\n",
       "          [ 6.4637e-01,  1.5261e+00, -2.9114e-02,  ..., -9.3400e-01,\n",
       "            1.0313e+00, -6.6548e-01]],\n",
       "\n",
       "         [[-4.3540e-01, -2.1627e+00, -8.4861e-01,  ..., -1.0322e-01,\n",
       "           -3.5110e-01, -2.0418e+00],\n",
       "          [-5.8553e-01, -1.0512e+00,  6.2625e-01,  ..., -7.2138e-01,\n",
       "            5.2855e-01, -7.2981e-01],\n",
       "          [-2.6224e-01, -5.8590e-01,  7.9889e-02,  ...,  6.3054e-01,\n",
       "           -3.0197e-01, -3.2926e-01],\n",
       "          ...,\n",
       "          [-1.8163e-01, -6.2936e-01,  3.8486e-02,  ..., -5.3871e-02,\n",
       "            2.2199e-01, -1.0439e+00],\n",
       "          [ 4.4829e-01, -5.5606e-01, -4.4620e-01,  ..., -2.1518e-01,\n",
       "           -4.1332e-03, -1.0793e+00],\n",
       "          [ 1.1344e+00, -1.5683e+00,  3.9334e-01,  ..., -7.2918e-01,\n",
       "            7.5417e-01, -3.3256e-01]],\n",
       "\n",
       "         [[-5.4207e-01, -3.5088e-01, -2.8063e-01,  ..., -9.8582e-01,\n",
       "            4.4790e-01, -5.4879e-01],\n",
       "          [ 1.7384e+00,  1.0848e+00,  2.1152e-01,  ..., -7.4697e-01,\n",
       "           -3.0102e-01,  1.2289e+00],\n",
       "          [ 1.5949e-01, -1.1023e-01,  1.3855e-01,  ...,  1.1258e-01,\n",
       "            1.9888e-01,  3.7988e-01],\n",
       "          ...,\n",
       "          [ 4.1116e-01, -6.6944e-02, -1.0216e+00,  ...,  2.3088e-01,\n",
       "            5.4524e-01,  5.5216e-01],\n",
       "          [-6.3590e-01, -5.8118e-01, -1.0704e+00,  ...,  2.6524e-02,\n",
       "            4.2748e-01,  7.9940e-02],\n",
       "          [ 1.7706e-01,  5.7449e-01, -8.6585e-02,  ..., -6.7468e-01,\n",
       "            3.3532e-01,  5.5338e-02]]],\n",
       "\n",
       "\n",
       "        [[[-3.7482e-01, -6.2762e-02,  2.6245e-01,  ..., -1.8246e-01,\n",
       "           -4.3347e-01,  2.2770e-01],\n",
       "          [ 2.6469e-01,  1.5745e-01, -4.2648e-01,  ..., -1.9641e-01,\n",
       "           -2.6379e-01, -2.4088e-02],\n",
       "          [ 4.7627e-01,  3.7798e-01, -7.6719e-01,  ..., -4.5478e-01,\n",
       "           -2.8487e-02,  4.1262e-01],\n",
       "          ...,\n",
       "          [-1.1655e-01,  7.2351e-02,  6.1625e-02,  ...,  8.6876e-02,\n",
       "           -3.2048e-01,  5.0459e-01],\n",
       "          [ 3.0638e-01,  6.3409e-01, -1.3872e-01,  ..., -1.4737e-01,\n",
       "           -4.0890e-01,  4.1711e-01],\n",
       "          [-7.8715e-02,  1.1463e+00, -6.6878e-02,  ..., -8.1114e-02,\n",
       "            5.4410e-01,  2.4319e-01]],\n",
       "\n",
       "         [[ 1.1592e-01,  7.5142e-01,  3.8636e-01,  ...,  2.7971e-01,\n",
       "            6.1046e-03,  5.0557e-01],\n",
       "          [-3.4326e-01,  1.8805e-01,  1.1985e-01,  ...,  9.4518e-02,\n",
       "            4.5554e-01,  4.2930e-02],\n",
       "          [-5.3385e-01,  4.8483e-01, -1.1393e-01,  ..., -4.3294e-02,\n",
       "            2.3888e-01, -5.1996e-02],\n",
       "          ...,\n",
       "          [-1.0038e-01, -4.9300e-01,  3.8508e-01,  ..., -4.6439e-01,\n",
       "            1.0606e+00, -3.1505e-01],\n",
       "          [-1.7191e-01,  2.8653e-01, -5.0431e-01,  ..., -2.2720e-03,\n",
       "            1.9717e-02, -5.1260e-02],\n",
       "          [ 6.5337e-01,  4.1585e-01, -3.5737e-01,  ...,  1.2655e-01,\n",
       "            4.0172e-01, -2.1383e-01]],\n",
       "\n",
       "         [[-1.1134e+00, -4.4958e-01, -8.3768e-01,  ..., -3.2334e-01,\n",
       "           -3.1784e-01, -2.1570e-01],\n",
       "          [-2.0843e-01, -1.0618e-01, -5.7915e-01,  ...,  4.7070e-02,\n",
       "           -6.6447e-01, -4.0470e-01],\n",
       "          [-7.6474e-01, -3.3219e-01, -8.4767e-01,  ...,  9.4391e-02,\n",
       "           -6.5666e-01,  1.5496e-01],\n",
       "          ...,\n",
       "          [-1.6590e-01, -6.1203e-01, -7.0388e-01,  ...,  7.9043e-02,\n",
       "           -2.3964e-01,  2.6050e-01],\n",
       "          [-1.1582e+00, -7.3130e-01, -5.1691e-01,  ..., -5.2298e-01,\n",
       "           -5.9170e-01, -4.4866e-02],\n",
       "          [-1.2179e+00, -5.7899e-01, -5.3422e-01,  ..., -3.7686e-01,\n",
       "            1.0230e+00,  8.6062e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.1677e-01,  9.2615e-01,  8.5153e-02,  ..., -4.8126e-01,\n",
       "            2.4995e-01, -2.6533e-01],\n",
       "          [-8.1273e-02,  6.4482e-01, -4.5910e-01,  ...,  1.3828e-01,\n",
       "           -2.4346e-02, -3.1649e-01],\n",
       "          [ 1.5670e-01,  1.1571e+00, -1.3332e-01,  ...,  2.8614e-02,\n",
       "            9.6297e-02, -2.8030e-01],\n",
       "          ...,\n",
       "          [ 1.1268e-01,  5.1285e-01, -4.2164e-01,  ..., -1.5636e-01,\n",
       "           -7.2862e-01,  2.6443e-02],\n",
       "          [ 5.5760e-01,  1.0591e+00, -2.1280e-01,  ..., -1.0574e-01,\n",
       "            2.6036e-01,  1.2971e-02],\n",
       "          [ 5.5704e-01,  8.1237e-01, -9.4767e-01,  ..., -5.4371e-01,\n",
       "            1.7246e+00, -7.6019e-01]],\n",
       "\n",
       "         [[-4.3396e-01, -2.2080e+00, -8.6459e-01,  ..., -9.5443e-02,\n",
       "           -3.5074e-01, -2.0449e+00],\n",
       "          [-1.7841e-01, -1.1480e+00, -1.3009e-01,  ...,  5.6083e-01,\n",
       "           -4.9602e-01, -1.0169e+00],\n",
       "          [ 1.9511e-01, -1.4096e+00, -4.5307e-01,  ..., -2.5735e-03,\n",
       "           -3.7215e-01, -1.3646e+00],\n",
       "          ...,\n",
       "          [ 4.9565e-01, -7.5667e-01, -2.2234e-01,  ...,  1.2309e+00,\n",
       "           -5.3332e-02, -4.9564e-01],\n",
       "          [ 2.1213e-01, -1.1459e+00, -4.4049e-01,  ..., -1.9521e-01,\n",
       "            4.2355e-01, -7.8724e-01],\n",
       "          [ 1.1616e+00, -1.8698e+00,  1.0286e+00,  ..., -4.9135e-01,\n",
       "            4.7393e-01, -1.7702e-01]],\n",
       "\n",
       "         [[-5.1282e-01, -3.7957e-01, -2.5469e-01,  ..., -1.0010e+00,\n",
       "            4.8779e-01, -5.6483e-01],\n",
       "          [-1.6966e-01, -2.7630e-01,  3.7451e-01,  ...,  1.6896e-01,\n",
       "           -5.4748e-02, -3.7120e-01],\n",
       "          [-4.2720e-01, -4.7217e-01,  4.0662e-01,  ..., -5.6136e-01,\n",
       "            1.0670e-01, -2.4474e-02],\n",
       "          ...,\n",
       "          [-4.1462e-01, -2.7937e-01, -6.6073e-02,  ...,  5.7552e-02,\n",
       "           -2.7460e-01,  2.7503e-01],\n",
       "          [-5.0214e-01, -1.4757e-01,  4.0340e-01,  ..., -7.0436e-01,\n",
       "            4.6472e-01,  2.4868e-01],\n",
       "          [-2.3069e-02,  9.0567e-01, -5.4004e-01,  ..., -8.7971e-01,\n",
       "           -6.1500e-01,  2.6350e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 1.2439e-02,  3.7830e-01,  8.3719e-01,  ...,  1.3049e+00,\n",
       "            3.3837e-02, -3.8861e-01],\n",
       "          [-8.2324e-01,  7.4628e-01, -2.3551e+00,  ...,  1.5080e+00,\n",
       "           -1.2701e+00, -3.2877e-01],\n",
       "          [ 8.0052e-01,  8.7527e-01,  5.6280e-01,  ...,  1.1330e+00,\n",
       "           -1.8750e+00, -1.2638e+00],\n",
       "          ...,\n",
       "          [-1.8874e-01,  1.7286e+00,  1.2884e+00,  ..., -1.5877e+00,\n",
       "           -2.5506e+00, -2.0636e+00],\n",
       "          [ 1.1398e-01,  1.9443e+00,  3.2200e+00,  ...,  2.5454e-01,\n",
       "           -1.0037e+00,  1.0599e+00],\n",
       "          [-3.9040e-01,  1.9833e+00,  5.0288e+00,  ..., -2.3369e+00,\n",
       "           -1.2608e+00,  1.0367e+00]],\n",
       "\n",
       "         [[ 3.3373e-01, -3.7880e-01,  4.1987e-01,  ...,  6.0616e-01,\n",
       "            2.7657e+00, -1.1110e+00],\n",
       "          [-2.3425e-01, -2.4043e+00, -1.7389e+00,  ...,  3.6978e+00,\n",
       "            1.7011e+00, -2.9644e+00],\n",
       "          [-1.5312e+00,  2.5463e-01, -1.1081e+00,  ...,  3.1860e+00,\n",
       "            1.5993e+00, -1.1313e-01],\n",
       "          ...,\n",
       "          [-3.5468e-01, -1.6214e+00, -1.0056e+00,  ...,  1.7368e+00,\n",
       "           -3.0073e-01, -5.4772e-01],\n",
       "          [ 1.9460e-01,  1.7992e-01,  1.0945e+00,  ..., -6.1372e-03,\n",
       "            2.5918e+00,  1.5825e+00],\n",
       "          [ 8.2528e-01,  4.0657e-01,  1.9142e+00,  ...,  2.0669e+00,\n",
       "            8.7231e-01, -1.1006e+00]],\n",
       "\n",
       "         [[ 8.7837e-01,  3.2679e+00,  4.0357e-01,  ...,  3.0024e+00,\n",
       "            7.6861e-02, -5.5741e-01],\n",
       "          [ 1.0043e+00,  3.1060e+00,  2.2265e+00,  ...,  7.1791e-01,\n",
       "            7.8478e-02,  8.9511e-01],\n",
       "          [-1.2746e+00,  1.9377e+00,  1.9651e+00,  ...,  1.4112e+00,\n",
       "           -2.9055e-01, -6.9861e-01],\n",
       "          ...,\n",
       "          [ 6.3226e-02,  1.3826e+00,  2.6932e+00,  ...,  2.1051e+00,\n",
       "           -2.1172e+00, -2.0158e+00],\n",
       "          [ 4.0871e-01,  6.8086e-01, -4.7026e-01,  ...,  7.8399e-01,\n",
       "            4.9957e-01, -5.4696e-01],\n",
       "          [ 9.0208e-01,  8.1596e-01,  1.4729e+00,  ..., -1.9745e+00,\n",
       "            1.5971e+00,  1.3855e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.8621e-01,  1.9061e+00, -8.1360e-01,  ...,  7.0112e-01,\n",
       "           -1.1144e+00,  1.8399e+00],\n",
       "          [ 3.5818e+00,  2.0852e+00, -2.3477e+00,  ...,  1.9771e-01,\n",
       "           -1.3881e+00,  7.2715e-01],\n",
       "          [ 1.4568e-01,  8.3456e-01,  1.5635e+00,  ..., -2.0982e+00,\n",
       "            1.3003e+00, -5.1965e-01],\n",
       "          ...,\n",
       "          [ 1.1368e-01,  6.0535e-01, -1.8018e+00,  ..., -2.3545e+00,\n",
       "            1.0827e+00, -8.8353e-01],\n",
       "          [ 1.9027e+00,  6.2439e-01, -1.4992e+00,  ..., -1.3074e-01,\n",
       "            2.1960e-01,  3.6762e-01],\n",
       "          [-3.8606e+00,  3.2153e+00, -1.0899e+00,  ...,  4.8345e+00,\n",
       "            1.3093e+00,  1.4719e+00]],\n",
       "\n",
       "         [[ 1.5187e+00, -2.5950e-01,  1.5249e+00,  ..., -9.9437e-01,\n",
       "            8.3837e-01, -2.1179e+00],\n",
       "          [ 1.2639e+00,  9.9099e-02, -2.5741e-01,  ...,  1.0914e-01,\n",
       "           -3.8661e-01,  1.9446e+00],\n",
       "          [ 1.7063e+00,  9.2033e-01, -4.3471e+00,  ...,  3.2567e-01,\n",
       "           -5.1035e+00, -2.6909e-01],\n",
       "          ...,\n",
       "          [ 1.9173e+00,  7.5307e-01,  2.1111e+00,  ...,  2.2781e+00,\n",
       "           -4.6421e+00,  6.5681e-01],\n",
       "          [ 7.3634e-01,  1.0006e+00, -5.3329e-01,  ...,  2.4505e+00,\n",
       "           -1.9811e+00, -1.7037e+00],\n",
       "          [ 1.0699e-01,  4.4506e+00,  1.2274e+00,  ..., -7.0878e-02,\n",
       "           -2.9136e+00,  2.4184e+00]],\n",
       "\n",
       "         [[ 2.9401e+00,  8.7464e-01,  5.2749e-01,  ...,  2.4670e+00,\n",
       "           -8.2950e-01, -1.4462e+00],\n",
       "          [-2.9342e-02,  8.4179e-01, -9.7711e-02,  ..., -3.2712e-01,\n",
       "           -1.3457e+00, -2.4749e+00],\n",
       "          [ 1.2154e-01,  7.8259e-02,  1.3675e+00,  ..., -1.2367e+00,\n",
       "           -2.5873e+00, -1.8732e+00],\n",
       "          ...,\n",
       "          [ 1.1113e+00, -1.6903e+00,  6.6623e-01,  ..., -7.9420e-01,\n",
       "            1.4711e-01, -1.3504e+00],\n",
       "          [-8.4288e-01,  1.7200e-01, -2.1180e+00,  ..., -5.8469e-01,\n",
       "           -7.1330e-01, -2.3023e+00],\n",
       "          [ 1.1535e+00, -2.2115e+00, -8.2452e-01,  ..., -5.9515e-01,\n",
       "           -3.9387e-01,  7.0475e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 6.0461e-01, -2.6529e-01,  1.7657e+00,  ..., -2.4689e+00,\n",
       "            2.3901e+00, -2.1815e-01],\n",
       "          [ 1.3340e-01,  4.7102e-01,  2.5479e+00,  ..., -6.1078e-01,\n",
       "           -1.1624e+00, -1.3834e+00],\n",
       "          [-2.3772e+00, -8.1985e-01,  1.6164e+00,  ..., -5.4298e-01,\n",
       "           -5.1411e-01, -1.4755e+00],\n",
       "          ...,\n",
       "          [-1.1122e+00,  2.6114e-01,  9.2983e-01,  ...,  1.2666e+00,\n",
       "            2.3124e+00,  1.9603e+00],\n",
       "          [-4.0038e-01,  2.2911e-01,  1.4164e+00,  ..., -3.6086e-01,\n",
       "            1.8533e+00,  1.4127e+00],\n",
       "          [-4.3223e-01,  2.0167e+00,  5.0823e+00,  ..., -2.4333e+00,\n",
       "           -1.2265e+00,  1.0898e+00]],\n",
       "\n",
       "         [[-2.7136e+00, -2.4510e+00, -1.4319e+00,  ...,  5.4014e-02,\n",
       "            1.1362e+00,  8.9236e-02],\n",
       "          [-1.5180e+00,  7.2373e-01,  6.0310e-01,  ...,  2.1199e+00,\n",
       "            2.0934e+00,  1.0474e+00],\n",
       "          [ 1.9137e+00, -7.6747e-01,  7.0731e-01,  ...,  2.1101e+00,\n",
       "            1.9363e+00,  1.7215e+00],\n",
       "          ...,\n",
       "          [ 1.8130e+00, -1.3937e-01, -3.6365e-01,  ..., -3.2507e-01,\n",
       "            9.6644e-01, -3.4332e+00],\n",
       "          [-1.0220e+00, -2.2834e-01, -8.2271e-01,  ..., -8.3600e-01,\n",
       "            2.6929e+00, -2.8267e+00],\n",
       "          [ 7.5408e-01,  4.4164e-01,  1.8761e+00,  ...,  2.0817e+00,\n",
       "            9.9783e-01, -1.2525e+00]],\n",
       "\n",
       "         [[ 8.9367e-01,  1.0896e+00,  1.9996e-01,  ...,  2.9177e+00,\n",
       "           -8.1836e-01, -2.2472e-01],\n",
       "          [-7.4977e-01,  2.3233e+00,  5.9976e-01,  ...,  2.1210e+00,\n",
       "           -5.9885e-01, -5.3025e-01],\n",
       "          [-1.0393e+00,  1.1830e+00,  7.2689e-02,  ...,  1.1009e+00,\n",
       "            1.8118e+00, -1.8751e+00],\n",
       "          ...,\n",
       "          [-1.9511e+00, -1.6631e-01,  1.3622e+00,  ...,  1.3708e+00,\n",
       "            2.2311e+00, -1.0283e+00],\n",
       "          [ 9.6541e-01,  1.5980e+00,  4.8686e-01,  ...,  1.0833e+00,\n",
       "            9.7043e-01, -2.2604e-01],\n",
       "          [ 9.1287e-01,  8.4441e-01,  1.5708e+00,  ..., -1.9849e+00,\n",
       "            1.5985e+00,  1.4041e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.1314e+00,  8.5108e-02, -2.1788e+00,  ..., -4.7437e-01,\n",
       "            1.0950e+00, -6.9593e-01],\n",
       "          [-7.8883e-01,  1.6687e-01,  7.2366e-01,  ..., -4.0958e+00,\n",
       "           -1.0363e+00,  4.6453e-01],\n",
       "          [-7.9050e-01,  2.8694e-01, -2.9486e-01,  ..., -2.8405e+00,\n",
       "            2.2372e+00,  9.0834e-01],\n",
       "          ...,\n",
       "          [ 2.3099e+00,  2.2785e+00,  2.1176e+00,  ...,  9.8745e-01,\n",
       "            7.4965e-01, -2.2285e+00],\n",
       "          [ 7.2364e-01, -1.1034e-01, -1.1224e-01,  ...,  2.0372e+00,\n",
       "           -3.2682e-01,  1.8598e+00],\n",
       "          [-3.8108e+00,  3.2063e+00, -1.0468e+00,  ...,  4.9505e+00,\n",
       "            1.2986e+00,  1.4698e+00]],\n",
       "\n",
       "         [[ 1.4904e+00, -3.2041e-01, -4.0492e-01,  ..., -1.0002e-01,\n",
       "           -2.1753e+00,  6.6779e-01],\n",
       "          [ 1.9388e+00,  1.1185e+00, -3.7252e+00,  ...,  2.1013e+00,\n",
       "           -2.7038e+00, -2.0760e+00],\n",
       "          [ 4.5996e-01,  2.1288e+00, -1.8797e+00,  ..., -3.3784e-01,\n",
       "            1.7861e+00, -3.7094e-01],\n",
       "          ...,\n",
       "          [ 1.5318e+00,  1.2378e+00,  1.6254e+00,  ..., -1.5972e+00,\n",
       "           -1.3227e+00,  3.1377e+00],\n",
       "          [ 1.2593e-01, -1.9418e-01,  2.9123e+00,  ...,  1.5818e+00,\n",
       "           -1.5466e+00,  1.5175e+00],\n",
       "          [ 4.5842e-02,  4.4280e+00,  1.2821e+00,  ..., -1.2710e-01,\n",
       "           -2.9599e+00,  2.5607e+00]],\n",
       "\n",
       "         [[ 3.3456e+00, -7.7396e-01,  9.4287e-01,  ...,  1.4205e+00,\n",
       "           -4.8154e-02, -2.7211e+00],\n",
       "          [ 5.6216e-01, -4.3953e-01,  1.8407e+00,  ..., -5.3926e-01,\n",
       "           -2.3562e+00, -1.6135e+00],\n",
       "          [ 1.7780e+00,  2.0801e+00,  1.8731e+00,  ...,  2.7112e-01,\n",
       "           -1.7597e+00, -1.9621e+00],\n",
       "          ...,\n",
       "          [ 7.3116e-01, -1.4470e+00,  1.0783e+00,  ..., -9.4711e-01,\n",
       "            2.8603e+00,  2.1330e+00],\n",
       "          [-1.4159e+00, -1.1879e-01, -1.6788e+00,  ..., -6.2542e-02,\n",
       "            4.9103e-02, -4.7889e-01],\n",
       "          [ 1.1899e+00, -2.2231e+00, -7.4631e-01,  ..., -5.8494e-01,\n",
       "           -3.0870e-01,  6.8425e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 5.2650e-01, -3.7911e+00,  3.3886e+00,  ..., -1.7034e+00,\n",
       "            2.0021e-01, -2.7179e+00],\n",
       "          [ 5.1722e-01, -2.1378e+00,  1.7640e+00,  ..., -5.2609e-01,\n",
       "           -9.2270e-01, -1.1043e+00],\n",
       "          [ 1.4302e+00,  1.5444e+00,  1.5654e+00,  ...,  2.2739e-01,\n",
       "           -1.8965e+00,  1.7517e+00],\n",
       "          ...,\n",
       "          [ 9.2839e-01,  9.3887e-01,  4.4973e+00,  ..., -8.9223e-01,\n",
       "           -4.1327e-01, -1.3577e+00],\n",
       "          [ 9.1505e-02,  6.0438e-01,  2.1076e+00,  ..., -2.3968e-01,\n",
       "           -7.2504e-01,  2.2628e-01],\n",
       "          [-3.7134e-01,  1.9826e+00,  5.0632e+00,  ..., -2.4443e+00,\n",
       "           -1.2520e+00,  1.0753e+00]],\n",
       "\n",
       "         [[-1.0013e+00, -2.5058e+00,  2.1257e+00,  ...,  2.8220e-01,\n",
       "            3.4706e-01, -9.8385e-01],\n",
       "          [-1.3188e+00,  8.3461e-01,  1.0207e+00,  ...,  4.4162e+00,\n",
       "            1.0869e+00,  2.2390e+00],\n",
       "          [-2.5724e+00, -2.6094e+00, -1.0081e+00,  ...,  2.1518e+00,\n",
       "            4.0445e+00, -7.7452e-01],\n",
       "          ...,\n",
       "          [-1.9845e+00, -8.4581e-01,  1.2602e+00,  ..., -2.2618e+00,\n",
       "            1.5260e+00,  1.1834e+00],\n",
       "          [ 7.1640e-01, -2.5614e-01,  8.6511e-01,  ..., -1.0802e+00,\n",
       "           -2.6820e-01, -9.0868e-01],\n",
       "          [ 7.9296e-01,  4.0697e-01,  1.8376e+00,  ...,  2.0797e+00,\n",
       "            9.5232e-01, -1.1100e+00]],\n",
       "\n",
       "         [[-2.6591e+00,  1.1630e+00,  2.5176e+00,  ...,  5.9868e-01,\n",
       "           -2.0724e+00,  7.6628e-01],\n",
       "          [ 4.5214e-01,  2.2477e+00,  3.1937e-01,  ...,  3.1157e+00,\n",
       "           -5.6327e-01, -6.3295e-01],\n",
       "          [-1.6631e-01,  3.2178e+00,  1.2648e+00,  ...,  4.8673e+00,\n",
       "           -2.0183e+00,  5.2334e-01],\n",
       "          ...,\n",
       "          [ 4.9363e-01, -3.6081e-01,  1.8606e-01,  ...,  2.8688e+00,\n",
       "           -4.5872e+00,  8.0608e-01],\n",
       "          [-1.2568e-01,  2.5957e-02, -3.1146e-01,  ...,  2.7265e+00,\n",
       "           -4.0056e-01,  1.7938e+00],\n",
       "          [ 8.8796e-01,  8.6319e-01,  1.4728e+00,  ..., -1.9420e+00,\n",
       "            1.5891e+00,  1.4452e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.8236e-01,  8.0662e-01, -1.3352e+00,  ..., -1.6463e+00,\n",
       "            2.8747e+00, -1.9095e+00],\n",
       "          [-1.3987e+00,  2.3867e-01,  4.1745e-01,  ..., -1.9027e+00,\n",
       "            2.6573e+00, -6.9105e-01],\n",
       "          [ 2.2940e+00, -3.8114e-01, -4.5298e-01,  ..., -2.5979e+00,\n",
       "           -7.2862e+00,  2.7087e+00],\n",
       "          ...,\n",
       "          [ 1.5265e-01,  1.1437e+00, -3.7778e+00,  ...,  3.8829e+00,\n",
       "           -2.3862e+00, -1.8682e+00],\n",
       "          [ 3.6612e+00, -1.3981e+00, -1.1636e+00,  ..., -1.6797e+00,\n",
       "           -1.1310e+00,  1.1167e+00],\n",
       "          [-3.8727e+00,  3.1309e+00, -1.0480e+00,  ...,  4.8780e+00,\n",
       "            1.3129e+00,  1.4892e+00]],\n",
       "\n",
       "         [[ 8.7462e-01,  9.7696e-01,  2.9515e+00,  ..., -1.7712e+00,\n",
       "           -5.1455e-02,  3.7186e-01],\n",
       "          [-6.9641e-01, -2.0973e-02, -1.5685e+00,  ..., -1.3282e+00,\n",
       "           -3.2463e+00, -1.2272e+00],\n",
       "          [-9.9177e-01, -1.6380e+00,  4.8620e+00,  ...,  7.7665e+00,\n",
       "            2.7704e+00, -3.6917e+00],\n",
       "          ...,\n",
       "          [ 1.3492e+00,  3.1525e+00,  8.6019e-01,  ...,  4.2139e+00,\n",
       "            1.5078e+00, -5.6034e-01],\n",
       "          [ 3.0123e-01,  4.9449e-01,  6.9919e-01,  ...,  3.1511e-01,\n",
       "           -2.5965e-01, -9.2297e-01],\n",
       "          [ 6.7331e-02,  4.4888e+00,  1.2454e+00,  ..., -1.1054e-01,\n",
       "           -2.9047e+00,  2.4208e+00]],\n",
       "\n",
       "         [[ 1.3232e-01,  1.1926e+00,  6.8295e-01,  ...,  2.2377e+00,\n",
       "           -1.7778e+00, -9.6630e-01],\n",
       "          [ 1.4864e+00,  6.3688e-01,  1.4264e+00,  ...,  6.7108e-01,\n",
       "           -3.8965e-01, -2.5003e+00],\n",
       "          [ 3.1211e+00, -1.1442e+00, -4.9709e-01,  ..., -2.7895e-01,\n",
       "           -1.3370e+00, -3.0905e+00],\n",
       "          ...,\n",
       "          [ 2.1258e-01, -2.3770e+00, -7.6275e-01,  ..., -2.5166e+00,\n",
       "            2.4570e-01, -7.7960e-01],\n",
       "          [ 1.0892e+00, -7.1109e-01, -3.0740e+00,  ...,  4.8735e-02,\n",
       "            7.4137e-01, -2.3371e+00],\n",
       "          [ 1.2383e+00, -2.2479e+00, -8.3630e-01,  ..., -5.8297e-01,\n",
       "           -4.3695e-01,  6.8031e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-8.4086e-01, -3.0563e-01,  1.7184e+00,  ...,  1.1802e+00,\n",
       "           -3.3156e-01,  1.6880e-01],\n",
       "          [ 1.0397e-01, -1.8632e+00,  4.2942e-01,  ...,  2.8879e+00,\n",
       "           -9.5676e-01, -7.4592e-01],\n",
       "          [ 3.5017e-01, -1.5654e-01,  2.1746e+00,  ...,  2.3524e+00,\n",
       "           -1.4253e+00, -1.7745e-01],\n",
       "          ...,\n",
       "          [ 4.9994e-02, -1.3203e+00,  5.4606e-01,  ...,  1.4553e+00,\n",
       "           -1.6527e+00, -1.7611e+00],\n",
       "          [-4.7199e-01, -1.1229e+00,  1.3216e+00,  ...,  1.1264e+00,\n",
       "           -9.8705e-01, -7.6881e-02],\n",
       "          [ 2.0188e-01,  8.1966e-02,  3.4887e-01,  ...,  3.2559e-01,\n",
       "           -2.9791e-01,  1.2333e-01]],\n",
       "\n",
       "         [[ 1.0316e-01, -1.9534e-03, -1.2553e+00,  ..., -1.4641e+00,\n",
       "            2.8526e-01, -5.0190e-01],\n",
       "          [-1.1516e+00, -5.8920e-01,  9.6716e-01,  ...,  1.8645e+00,\n",
       "           -7.5916e-02, -7.8767e-01],\n",
       "          [-3.0662e-01, -1.9833e-01, -4.0854e-01,  ...,  1.5755e+00,\n",
       "            5.6741e-01,  7.0511e-01],\n",
       "          ...,\n",
       "          [-1.5330e+00, -3.7664e-01,  8.3821e-01,  ...,  2.6253e-01,\n",
       "           -8.8878e-01, -7.2165e-01],\n",
       "          [-5.5608e-01,  3.5628e-02, -4.0589e-01,  ...,  4.0209e-01,\n",
       "           -1.0446e+00,  6.5316e-01],\n",
       "          [ 5.8705e-02, -4.4675e-01, -5.7710e-01,  ...,  3.7641e-01,\n",
       "           -4.2803e-01, -1.4676e-01]],\n",
       "\n",
       "         [[ 5.6327e-01,  6.3299e-01, -1.0600e+00,  ...,  1.4695e+00,\n",
       "           -1.6861e+00, -1.6401e+00],\n",
       "          [ 1.2321e-01,  1.4024e+00,  1.2007e-01,  ...,  1.7546e+00,\n",
       "           -1.5474e+00, -1.7478e+00],\n",
       "          [ 2.5359e-01,  2.9003e-01,  9.1222e-01,  ...,  1.0413e+00,\n",
       "           -1.3417e+00, -1.2838e+00],\n",
       "          ...,\n",
       "          [-1.0390e+00,  2.3952e+00,  1.6541e+00,  ...,  1.2251e+00,\n",
       "           -7.6673e-01, -1.2747e+00],\n",
       "          [ 3.4970e-01,  4.2991e-01,  1.5271e+00,  ...,  4.2636e-01,\n",
       "           -1.0643e+00,  4.9321e-01],\n",
       "          [ 1.1749e-01,  9.1233e-01, -4.4913e-01,  ..., -7.0242e-01,\n",
       "            1.1672e-01,  7.7128e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.6987e+00, -8.8735e-01, -8.9475e-02,  ..., -2.5567e+00,\n",
       "            9.0129e-01,  1.4277e+00],\n",
       "          [-8.4368e-01,  6.9310e-01, -2.1558e-01,  ..., -1.1061e+00,\n",
       "           -1.0537e+00,  8.8460e-01],\n",
       "          [-1.9590e+00,  7.4159e-01,  1.6952e+00,  ..., -1.1073e+00,\n",
       "           -1.1582e+00,  6.2480e-01],\n",
       "          ...,\n",
       "          [-1.4146e+00,  7.9593e-01,  1.8753e+00,  ...,  3.7335e-01,\n",
       "            4.6961e-01,  5.4024e-01],\n",
       "          [-1.8002e+00,  7.7272e-01, -7.6360e-01,  ..., -3.1379e-01,\n",
       "           -2.9923e-01,  1.6481e-01],\n",
       "          [-1.2082e-01,  4.7788e-02, -1.8814e-01,  ...,  1.5822e-01,\n",
       "            8.0160e-02,  2.7424e-01]],\n",
       "\n",
       "         [[-1.3917e+00, -1.9069e+00, -1.7620e+00,  ..., -4.0610e-01,\n",
       "            1.3780e+00, -3.8869e-01],\n",
       "          [-6.9403e-01, -2.0051e-01, -1.7985e+00,  ..., -7.4837e-02,\n",
       "            2.4256e+00, -6.5537e-01],\n",
       "          [-2.5966e+00,  7.0341e-01, -1.0521e+00,  ..., -3.7012e-01,\n",
       "            6.8487e-01, -2.2544e+00],\n",
       "          ...,\n",
       "          [ 6.0748e-01,  3.0423e-01,  4.6731e-01,  ...,  1.8274e+00,\n",
       "            2.4950e+00,  6.1630e-01],\n",
       "          [ 3.4134e-02, -2.1207e-01, -1.8412e-01,  ..., -8.3872e-01,\n",
       "            1.1080e+00,  1.3862e+00],\n",
       "          [-1.1582e-01,  3.4951e-01, -7.5160e-02,  ..., -2.5728e-01,\n",
       "           -5.9190e-03, -8.6738e-01]],\n",
       "\n",
       "         [[-3.0960e-01, -7.7013e-01,  4.6031e-01,  ..., -1.3052e+00,\n",
       "           -4.4826e-01,  1.2632e+00],\n",
       "          [-8.0717e-01, -1.6215e+00,  1.8776e-01,  ..., -8.5842e-01,\n",
       "           -5.4714e-01,  3.2438e-01],\n",
       "          [-1.2684e+00, -3.5972e-03,  1.7727e+00,  ...,  8.4858e-02,\n",
       "           -1.1075e-01,  2.0121e-01],\n",
       "          ...,\n",
       "          [ 7.7007e-02,  4.7267e-02,  9.0872e-01,  ..., -3.4556e-01,\n",
       "           -1.3524e+00, -1.0719e+00],\n",
       "          [ 6.4330e-01, -1.0441e+00,  1.8063e+00,  ...,  1.0791e-01,\n",
       "            1.1616e-01,  3.6037e-02],\n",
       "          [-6.5233e-01, -1.1698e-01,  3.9279e-01,  ..., -5.9053e-01,\n",
       "            9.3605e-01, -2.0841e-01]]],\n",
       "\n",
       "\n",
       "        [[[-8.9457e-01, -2.6476e+00, -4.4799e-01,  ...,  1.4595e+00,\n",
       "            1.1004e+00, -8.7250e-01],\n",
       "          [ 5.0797e-01, -1.0346e+00,  1.1949e+00,  ...,  3.0157e+00,\n",
       "           -1.7009e+00,  2.4899e-01],\n",
       "          [-1.2804e+00, -2.9195e+00,  6.9483e-01,  ...,  1.1631e-01,\n",
       "           -1.7536e+00, -1.3296e+00],\n",
       "          ...,\n",
       "          [-1.4172e+00, -2.0276e+00,  4.6672e-01,  ...,  6.7138e-02,\n",
       "           -4.2741e-01, -9.2468e-02],\n",
       "          [-2.0875e+00, -1.2194e+00,  2.0688e-01,  ...,  8.2761e-01,\n",
       "            7.2805e-01, -9.8300e-01],\n",
       "          [ 1.7785e-01,  5.6405e-02,  3.5455e-01,  ...,  3.5435e-01,\n",
       "           -2.9417e-01,  8.5946e-02]],\n",
       "\n",
       "         [[-1.7067e+00,  8.0126e-01,  4.5070e-03,  ...,  1.1578e+00,\n",
       "            2.3099e-01,  7.0395e-01],\n",
       "          [-7.2863e-01,  2.5724e-01, -1.3414e+00,  ...,  9.7764e-01,\n",
       "            2.5865e-01,  7.1913e-01],\n",
       "          [-1.9274e+00, -5.3739e-01, -7.0926e-01,  ...,  1.2599e+00,\n",
       "            8.2104e-01, -2.5918e-01],\n",
       "          ...,\n",
       "          [-2.0796e-01,  5.4078e-01, -3.8319e-01,  ...,  3.4608e-01,\n",
       "            5.2723e-01, -4.6560e-01],\n",
       "          [ 1.0886e-01,  1.2027e+00, -4.4993e-01,  ..., -3.0443e-01,\n",
       "           -1.7158e+00,  1.7630e-01],\n",
       "          [ 9.7119e-02, -4.2939e-01, -5.9448e-01,  ...,  3.6499e-01,\n",
       "           -4.1087e-01, -1.3724e-01]],\n",
       "\n",
       "         [[-1.5731e-02,  1.2957e+00, -9.1056e-02,  ...,  1.1863e+00,\n",
       "           -2.1708e+00, -1.0619e+00],\n",
       "          [ 2.9234e-01,  1.9601e-01,  1.5083e+00,  ...,  8.7397e-01,\n",
       "           -1.2934e+00, -9.0370e-01],\n",
       "          [ 9.6348e-01,  1.0701e+00,  1.5685e+00,  ..., -8.8426e-01,\n",
       "           -3.9311e-01,  9.4236e-01],\n",
       "          ...,\n",
       "          [ 1.2276e+00,  1.7221e-01, -1.3493e+00,  ...,  1.5578e+00,\n",
       "           -1.3867e+00,  5.3867e-01],\n",
       "          [-4.7968e-01,  2.1708e-01, -4.1656e-01,  ..., -7.0570e-01,\n",
       "           -2.0623e-01,  1.5500e+00],\n",
       "          [ 1.1139e-01,  8.9919e-01, -5.1432e-01,  ..., -7.6462e-01,\n",
       "            1.7928e-01,  7.3559e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.1186e-01, -4.5536e-03, -7.1517e-02,  ..., -9.1708e-02,\n",
       "           -1.3344e+00,  1.5693e+00],\n",
       "          [-1.7409e+00,  9.1566e-01,  1.5830e+00,  ..., -6.5713e-01,\n",
       "           -1.6836e-01,  1.4385e+00],\n",
       "          [ 8.4068e-01,  1.8357e+00, -1.9422e+00,  ..., -9.7132e-01,\n",
       "           -3.8966e-01,  1.8021e+00],\n",
       "          ...,\n",
       "          [-1.2945e+00, -5.0387e-01,  1.5933e-01,  ..., -1.9691e+00,\n",
       "            5.3620e-01,  6.3525e-01],\n",
       "          [-5.8211e-01, -5.2640e-01, -3.6576e-02,  ..., -1.6763e-01,\n",
       "            3.2453e-01,  7.2343e-01],\n",
       "          [-1.5042e-01,  1.8590e-02, -1.0145e-01,  ...,  1.3092e-01,\n",
       "            1.2837e-01,  2.7542e-01]],\n",
       "\n",
       "         [[ 1.9164e+00,  1.2122e+00, -1.3384e+00,  ...,  7.5436e-01,\n",
       "            1.4121e+00,  4.2900e-01],\n",
       "          [-1.8745e+00,  2.8360e-01, -1.0372e+00,  ...,  8.6428e-01,\n",
       "            5.1596e-01, -2.5327e+00],\n",
       "          [-1.8830e-01,  3.4482e-01, -2.7613e-01,  ...,  1.2010e+00,\n",
       "            2.4469e+00, -9.0939e-01],\n",
       "          ...,\n",
       "          [ 6.9563e-01, -1.1033e+00, -1.3713e+00,  ...,  2.9353e-01,\n",
       "            5.9285e-01, -9.0312e-01],\n",
       "          [-5.6697e-01, -6.3250e-01, -2.4669e-01,  ..., -3.6591e-01,\n",
       "            2.1328e-01,  7.3284e-01],\n",
       "          [-7.7913e-02,  3.1242e-01, -6.8418e-02,  ..., -2.9310e-01,\n",
       "           -9.8173e-02, -8.7538e-01]],\n",
       "\n",
       "         [[ 5.0111e-01, -4.3092e-01,  5.6828e-02,  ...,  2.7020e-01,\n",
       "           -3.6914e-01, -2.2238e+00],\n",
       "          [-3.1909e-01, -1.5345e-01,  1.5736e+00,  ...,  7.7824e-01,\n",
       "           -1.9213e-01,  9.0397e-01],\n",
       "          [-8.3800e-01, -2.0367e+00, -3.5273e-01,  ...,  8.5833e-01,\n",
       "           -2.8127e-01,  1.5068e+00],\n",
       "          ...,\n",
       "          [-1.6940e+00, -1.5061e+00,  2.1854e+00,  ..., -8.1315e-01,\n",
       "           -5.9195e-01, -4.7773e-01],\n",
       "          [-1.9428e-01, -1.1663e+00,  1.0474e+00,  ..., -5.8962e-01,\n",
       "           -1.2152e-01, -7.6329e-01],\n",
       "          [-6.2913e-01, -6.9152e-02,  3.9282e-01,  ..., -6.0157e-01,\n",
       "            9.2209e-01, -2.6900e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.1922e-02, -8.1158e-01,  9.4305e-01,  ...,  1.6516e+00,\n",
       "            7.7891e-01,  7.4810e-01],\n",
       "          [ 4.2586e-01, -8.1132e-01,  3.7848e+00,  ...,  1.4591e+00,\n",
       "           -9.5628e-02, -5.2547e-01],\n",
       "          [-1.7766e-01, -9.2944e-01,  2.0367e+00,  ...,  3.8420e+00,\n",
       "           -1.7914e+00,  1.7394e+00],\n",
       "          ...,\n",
       "          [-1.0007e+00, -6.6643e-02,  3.3855e-01,  ..., -3.1658e-01,\n",
       "           -3.5061e-01, -3.2258e-01],\n",
       "          [-5.4930e-01, -1.4572e+00,  1.0157e+00,  ..., -4.6836e-02,\n",
       "           -2.4741e-01, -4.7020e-01],\n",
       "          [ 1.9342e-01,  6.6243e-02,  3.6409e-01,  ...,  3.5059e-01,\n",
       "           -2.5916e-01,  1.0694e-01]],\n",
       "\n",
       "         [[-8.8100e-01,  2.2893e+00,  5.5227e-01,  ..., -1.9267e-01,\n",
       "            4.0999e-01, -5.3963e-01],\n",
       "          [-9.6073e-01,  1.5021e+00, -7.9679e-01,  ...,  1.0310e+00,\n",
       "            1.0003e-01, -3.4317e-01],\n",
       "          [-2.6142e+00,  1.5512e+00, -1.8991e+00,  ..., -7.1202e-01,\n",
       "           -3.5428e-01, -1.2328e+00],\n",
       "          ...,\n",
       "          [-1.0958e+00,  6.1058e-01,  4.0574e-01,  ...,  3.6602e-02,\n",
       "            6.7157e-01,  6.9343e-01],\n",
       "          [-7.8512e-01,  3.2619e-01,  3.6471e-01,  ...,  5.9806e-01,\n",
       "           -9.7565e-01,  1.4993e-03],\n",
       "          [ 6.9504e-02, -4.2047e-01, -6.0429e-01,  ...,  3.8201e-01,\n",
       "           -4.3715e-01, -1.5354e-01]],\n",
       "\n",
       "         [[ 1.1246e+00,  9.5592e-01, -7.8316e-02,  ...,  6.2027e-02,\n",
       "           -2.4287e-01, -1.5704e-01],\n",
       "          [-1.2452e-01, -3.3043e-01,  1.0110e+00,  ..., -8.1628e-01,\n",
       "           -1.3089e+00, -1.3034e+00],\n",
       "          [ 1.2373e+00,  1.0953e+00,  1.3478e+00,  ..., -4.2677e-01,\n",
       "           -5.7293e-01,  2.0286e-01],\n",
       "          ...,\n",
       "          [ 6.4824e-01,  6.2796e-01,  6.0488e-01,  ..., -1.1937e+00,\n",
       "           -1.8483e+00,  1.2702e+00],\n",
       "          [-5.9689e-01,  7.3291e-01,  4.9039e-01,  ...,  6.9316e-02,\n",
       "           -2.0106e+00,  1.0006e-01],\n",
       "          [ 1.3332e-01,  8.9192e-01, -4.6987e-01,  ..., -7.5452e-01,\n",
       "            1.3666e-01,  7.3603e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.2410e+00, -4.9150e-01, -9.7343e-01,  ...,  8.6639e-02,\n",
       "            1.5249e+00,  1.9951e-01],\n",
       "          [-2.0424e+00, -1.4266e+00, -5.8134e-01,  ..., -1.4909e+00,\n",
       "            1.9783e+00, -2.3347e-01],\n",
       "          [-1.2072e+00, -3.0351e-02, -3.8446e-01,  ..., -1.3100e+00,\n",
       "           -3.6517e-01,  5.9911e-01],\n",
       "          ...,\n",
       "          [ 6.7420e-01,  1.6304e+00, -5.7150e-01,  ..., -1.0778e-01,\n",
       "            6.3938e-01,  1.4464e+00],\n",
       "          [-5.3402e-01, -3.5608e-01, -1.4060e+00,  ..., -5.3283e-01,\n",
       "            7.4861e-01,  6.3224e-01],\n",
       "          [-1.6440e-01,  1.5966e-02, -1.5112e-01,  ...,  1.5090e-01,\n",
       "            1.1621e-01,  2.6679e-01]],\n",
       "\n",
       "         [[-5.2460e-01,  1.5028e+00, -1.8890e+00,  ..., -1.9252e-01,\n",
       "            3.1459e-01, -1.0256e+00],\n",
       "          [ 6.4793e-01,  2.1821e+00, -8.3300e-01,  ...,  1.6678e+00,\n",
       "            1.2059e+00, -2.2993e+00],\n",
       "          [ 5.1866e-01,  3.1921e-02,  3.4437e-01,  ..., -3.2494e-01,\n",
       "            7.5576e-01, -2.3107e+00],\n",
       "          ...,\n",
       "          [ 2.3668e-01,  3.0048e+00, -2.6355e-01,  ..., -5.6326e-01,\n",
       "            1.0351e+00, -1.6449e+00],\n",
       "          [ 1.6302e-01, -8.0206e-01, -1.5316e-01,  ..., -8.0708e-01,\n",
       "            1.6273e+00,  6.3419e-01],\n",
       "          [-8.4866e-02,  3.4350e-01, -9.4899e-02,  ..., -2.8584e-01,\n",
       "           -2.2787e-02, -8.8303e-01]],\n",
       "\n",
       "         [[-5.5661e-01, -1.8411e-01,  2.9613e-01,  ...,  6.4605e-01,\n",
       "           -7.8190e-01,  1.7199e+00],\n",
       "          [ 1.0661e+00, -1.7669e+00, -8.2074e-01,  ..., -8.1172e-01,\n",
       "            1.9591e+00, -3.0322e-01],\n",
       "          [ 1.1170e+00, -1.3617e+00, -1.1943e+00,  ...,  6.4948e-01,\n",
       "            1.4675e+00,  2.1006e-01],\n",
       "          ...,\n",
       "          [ 1.2531e+00,  3.2201e-01,  5.4611e-02,  ...,  7.1197e-01,\n",
       "            4.6821e-01, -5.8946e-01],\n",
       "          [ 1.6807e+00, -1.5193e+00, -4.0560e-01,  ...,  1.0031e-01,\n",
       "            1.8039e+00,  1.1484e+00],\n",
       "          [-6.1969e-01, -9.5675e-02,  3.6944e-01,  ..., -5.9986e-01,\n",
       "            9.7057e-01, -1.8116e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[ 1.1506e+00, -4.9291e-01, -3.9819e-01,  ..., -1.6062e-01,\n",
       "           -1.7529e+00, -1.3331e+00],\n",
       "          [ 2.8235e-01,  4.2121e-01, -1.6820e-01,  ..., -1.5400e-01,\n",
       "           -2.2818e-01,  1.0633e-01],\n",
       "          [ 7.7686e-01,  5.3778e-01, -2.8983e-01,  ...,  4.6916e-01,\n",
       "           -2.0826e-01, -2.6377e-01],\n",
       "          ...,\n",
       "          [ 5.1972e-01,  6.8986e-01, -4.5597e-01,  ..., -9.0689e-02,\n",
       "            7.5458e-01, -1.2928e-01],\n",
       "          [ 1.3723e-01,  2.3146e-01, -5.9133e-01,  ...,  3.2069e-02,\n",
       "            1.7707e-01, -3.6215e-01],\n",
       "          [-4.0015e-01,  6.5620e-01,  2.6226e-01,  ..., -6.5509e-01,\n",
       "            7.3958e-01, -6.3681e-01]],\n",
       "\n",
       "         [[-1.1030e+00,  1.2568e+00, -1.4504e+00,  ..., -1.0441e-01,\n",
       "           -2.9708e-01,  4.2761e-02],\n",
       "          [-6.9291e-01,  3.1393e-01, -7.6959e-01,  ..., -6.7575e-01,\n",
       "           -1.0428e-01, -3.0118e-01],\n",
       "          [-1.5579e-01,  3.2537e-02, -1.8133e-01,  ..., -9.6506e-01,\n",
       "           -1.2014e-01, -1.0482e+00],\n",
       "          ...,\n",
       "          [-1.7210e+00,  1.2774e+00, -1.0793e-01,  ..., -1.8180e-01,\n",
       "           -3.4300e-01, -2.0266e-01],\n",
       "          [-8.4348e-01,  1.0860e+00,  1.5546e-02,  ..., -4.8128e-01,\n",
       "           -4.6949e-01,  4.4404e-01],\n",
       "          [-1.1070e+00,  1.4874e+00,  6.6014e-01,  ...,  4.8635e-01,\n",
       "           -1.2029e+00, -1.2101e-01]],\n",
       "\n",
       "         [[ 1.3133e-01, -4.1037e-01,  5.6259e-01,  ..., -4.9210e-01,\n",
       "           -4.8576e-01, -1.7708e-01],\n",
       "          [ 2.7480e-01, -5.0928e-01,  3.8527e-01,  ..., -1.0681e-01,\n",
       "           -3.1057e-01,  4.4578e-01],\n",
       "          [-9.8341e-02, -6.0299e-01,  4.9029e-01,  ...,  9.8833e-02,\n",
       "           -3.3059e-01,  3.6319e-01],\n",
       "          ...,\n",
       "          [ 2.2845e-01,  2.7584e-01, -2.0434e-02,  ..., -5.8629e-02,\n",
       "           -4.3992e-01, -1.9506e-01],\n",
       "          [ 3.1873e-01,  1.2316e-01, -2.1157e-01,  ..., -5.5272e-01,\n",
       "           -2.9074e-02,  3.2775e-01],\n",
       "          [ 8.0294e-02,  5.7019e-02,  7.1892e-01,  ..., -1.0376e+00,\n",
       "            3.9582e-01,  1.0693e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.7843e-01, -1.2575e+00,  4.4305e-02,  ...,  7.6628e-01,\n",
       "           -3.1498e-03, -1.1135e+00],\n",
       "          [ 7.0778e-02,  1.5957e-01, -7.8162e-02,  ..., -3.5043e-01,\n",
       "            8.7418e-01,  2.2688e-01],\n",
       "          [-4.4052e-01, -1.7708e-01, -2.5521e-01,  ..., -5.0169e-01,\n",
       "            2.7499e-01,  3.8639e-01],\n",
       "          ...,\n",
       "          [-7.1382e-02,  3.0293e-02, -2.6190e-01,  ..., -5.1705e-01,\n",
       "            1.0557e-01, -4.0486e-01],\n",
       "          [ 2.3588e-01, -4.7333e-01,  1.5298e-02,  ..., -1.0905e-01,\n",
       "           -1.8965e-02, -3.1688e-01],\n",
       "          [ 3.1896e-01, -3.2816e-01, -1.6521e+00,  ...,  4.7915e-01,\n",
       "            1.2016e+00, -1.2021e-01]],\n",
       "\n",
       "         [[ 1.1748e+00, -6.7858e-01,  5.9019e-01,  ...,  5.1479e-01,\n",
       "            3.0926e-01,  7.9820e-01],\n",
       "          [ 5.5861e-01, -5.4178e-01, -1.7977e-01,  ...,  1.1838e-01,\n",
       "            8.3497e-01, -4.0075e-01],\n",
       "          [ 1.5045e-01, -1.4977e-01, -9.8687e-02,  ..., -2.0600e-01,\n",
       "            1.0616e+00,  1.9913e-01],\n",
       "          ...,\n",
       "          [-5.0912e-03, -6.9826e-03,  5.4472e-01,  ...,  6.6000e-03,\n",
       "            1.0812e+00,  1.7855e-01],\n",
       "          [ 4.0859e-01, -4.1852e-01,  7.3140e-01,  ..., -6.1150e-01,\n",
       "            9.3880e-01,  5.2785e-01],\n",
       "          [-5.2623e-01,  5.2842e-01,  6.9877e-01,  ..., -3.3959e-01,\n",
       "            1.1610e+00,  3.5257e-01]],\n",
       "\n",
       "         [[-1.4154e-01,  1.3668e-01,  5.5145e-01,  ..., -7.5615e-01,\n",
       "            3.8397e-01,  1.8153e-02],\n",
       "          [ 1.8191e-01, -4.3659e-01, -9.9774e-01,  ..., -3.0200e-02,\n",
       "           -3.3493e-02,  1.4781e-01],\n",
       "          [ 2.0583e-01, -5.2219e-01, -4.7116e-01,  ..., -4.7674e-01,\n",
       "            1.7686e-01,  7.9434e-01],\n",
       "          ...,\n",
       "          [-6.2595e-01, -7.0107e-01, -8.0970e-01,  ..., -2.7415e-01,\n",
       "           -6.0944e-01, -5.4084e-01],\n",
       "          [-4.0806e-01, -9.4359e-01, -5.0508e-01,  ..., -1.2047e+00,\n",
       "           -3.2539e-01, -2.1142e-01],\n",
       "          [-7.2810e-03, -3.5433e-01, -7.0797e-01,  ...,  3.5277e-01,\n",
       "            9.4728e-02,  4.1108e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.1940e+00, -4.7340e-01, -3.6252e-01,  ..., -1.7001e-01,\n",
       "           -1.7501e+00, -1.3677e+00],\n",
       "          [ 5.2420e-01,  3.6860e-01,  1.7294e-01,  ...,  7.3504e-01,\n",
       "           -1.2539e-01, -7.9622e-01],\n",
       "          [ 1.4284e-01,  3.8709e-01, -2.5431e-01,  ...,  8.2869e-01,\n",
       "           -2.6340e-01, -2.0934e-01],\n",
       "          ...,\n",
       "          [ 1.0208e+00,  8.2190e-02,  5.4793e-02,  ...,  7.8294e-01,\n",
       "            4.5586e-02,  8.0572e-02],\n",
       "          [ 3.8724e-01,  8.0655e-01, -1.9705e-01,  ..., -9.9419e-02,\n",
       "           -2.3816e-01, -7.2530e-01],\n",
       "          [ 5.4197e-01,  5.5417e-01,  1.2185e-03,  ..., -2.1756e-01,\n",
       "            5.8860e-01, -6.8847e-01]],\n",
       "\n",
       "         [[-1.0988e+00,  1.3162e+00, -1.5045e+00,  ..., -9.4240e-02,\n",
       "           -3.1146e-01,  5.2353e-02],\n",
       "          [-2.2279e-01,  1.1647e+00, -1.1749e+00,  ...,  2.3724e-02,\n",
       "            3.3992e-01,  6.1669e-02],\n",
       "          [-4.0978e-01,  1.2801e+00, -4.3649e-01,  ...,  5.2493e-02,\n",
       "            4.5525e-01,  9.7389e-02],\n",
       "          ...,\n",
       "          [-7.7351e-01,  1.3339e+00, -4.9279e-01,  ..., -2.9035e-01,\n",
       "           -2.8892e-01, -4.8164e-01],\n",
       "          [-6.6902e-01,  1.4633e+00, -5.2326e-01,  ..., -2.0510e-01,\n",
       "           -8.1412e-01, -7.5339e-01],\n",
       "          [-5.8892e-01,  1.4356e+00, -2.2874e-01,  ...,  3.9795e-02,\n",
       "           -1.1241e+00,  2.6672e-01]],\n",
       "\n",
       "         [[ 1.2137e-01, -4.0899e-01,  5.4896e-01,  ..., -4.8470e-01,\n",
       "           -5.0882e-01, -1.8896e-01],\n",
       "          [ 1.2095e-01, -7.4285e-01,  4.1877e-01,  ..., -9.3273e-01,\n",
       "            6.6590e-01,  4.6959e-01],\n",
       "          [ 4.8959e-02, -1.6050e-01, -9.5732e-02,  ...,  8.5072e-02,\n",
       "           -3.6102e-01, -2.3912e-01],\n",
       "          ...,\n",
       "          [ 4.8701e-01, -4.5380e-02, -4.3567e-01,  ..., -2.4708e-01,\n",
       "            2.0741e-01, -8.9335e-02],\n",
       "          [ 3.9059e-01,  1.4596e-01,  9.6919e-01,  ..., -4.6282e-02,\n",
       "           -9.3228e-01,  6.0143e-02],\n",
       "          [ 4.4860e-01, -2.1678e-02,  6.1003e-01,  ..., -6.7964e-01,\n",
       "            5.0548e-02,  4.9305e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.9715e-01, -1.2586e+00,  3.3025e-02,  ...,  7.7554e-01,\n",
       "            1.9215e-03, -1.1144e+00],\n",
       "          [ 1.5751e+00,  5.4388e-01, -6.8124e-01,  ...,  4.2246e-01,\n",
       "            4.3049e-01, -3.0168e-01],\n",
       "          [-9.7085e-02,  1.6307e-01, -3.7443e-01,  ...,  1.3748e-01,\n",
       "            4.4432e-01,  3.0440e-01],\n",
       "          ...,\n",
       "          [ 4.6855e-01,  2.2112e-01, -3.5185e-01,  ...,  1.2834e-01,\n",
       "           -2.7732e-02, -3.6870e-01],\n",
       "          [ 2.2437e-01, -1.3691e+00, -2.4617e-05,  ...,  1.4339e-01,\n",
       "           -1.6737e-01, -4.9901e-01],\n",
       "          [ 1.9324e-01, -9.9217e-01, -3.6800e-01,  ...,  2.1563e-01,\n",
       "            7.5341e-01, -2.5184e-01]],\n",
       "\n",
       "         [[ 1.1519e+00, -7.0147e-01,  6.0943e-01,  ...,  5.3862e-01,\n",
       "            3.3975e-01,  7.9671e-01],\n",
       "          [ 8.0454e-01, -5.6556e-01,  3.2940e-01,  ..., -1.0524e-02,\n",
       "            2.6127e-01,  9.1100e-01],\n",
       "          [-1.4264e-01, -2.8391e-01, -7.8567e-02,  ..., -5.5470e-01,\n",
       "            1.3020e+00,  2.9710e-01],\n",
       "          ...,\n",
       "          [ 1.2452e-01,  3.1988e-01, -3.1673e-01,  ...,  5.0334e-01,\n",
       "            1.3604e+00, -1.8863e-01],\n",
       "          [ 6.1446e-01, -1.0813e+00,  7.9415e-01,  ..., -8.5531e-02,\n",
       "            7.5011e-01,  3.4664e-01],\n",
       "          [ 2.7338e-01, -1.0004e-01,  8.2877e-01,  ..., -1.1460e-01,\n",
       "            5.6544e-01,  4.6493e-01]],\n",
       "\n",
       "         [[-8.2621e-02,  1.3373e-01,  5.8436e-01,  ..., -7.1668e-01,\n",
       "            3.6267e-01, -4.1685e-02],\n",
       "          [ 4.9319e-01,  5.3307e-01, -8.7349e-02,  ...,  3.4457e-01,\n",
       "           -4.6859e-02, -8.0067e-01],\n",
       "          [ 4.7420e-01,  5.1548e-02, -5.3723e-01,  ..., -9.1428e-02,\n",
       "           -2.1167e-01, -1.9809e-01],\n",
       "          ...,\n",
       "          [ 9.3158e-01, -3.3572e-01, -4.0132e-01,  ..., -4.9336e-01,\n",
       "           -5.8035e-01, -7.7478e-01],\n",
       "          [-7.2576e-02, -1.5505e-01,  1.5133e-01,  ..., -4.4788e-01,\n",
       "           -2.8136e-01, -8.9842e-01],\n",
       "          [-3.6189e-01, -2.3796e-01, -2.0025e-02,  ..., -2.5256e-02,\n",
       "            1.5153e-01, -2.8092e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.1996e+00, -4.8291e-01, -3.6647e-01,  ..., -1.7493e-01,\n",
       "           -1.7798e+00, -1.3480e+00],\n",
       "          [ 2.7634e-01,  1.9505e-01, -7.9814e-02,  ...,  2.5236e-01,\n",
       "           -1.8321e-01, -5.2510e-01],\n",
       "          [ 4.9775e-01,  6.1405e-02, -5.5239e-02,  ...,  4.1120e-02,\n",
       "           -2.1951e-02, -3.9503e-01],\n",
       "          ...,\n",
       "          [ 4.3437e-01, -1.4119e-01, -2.0721e-01,  ...,  1.8330e-01,\n",
       "           -5.1723e-03, -5.8115e-01],\n",
       "          [ 7.5523e-01,  2.4726e-01, -1.3121e-02,  ..., -2.5937e-01,\n",
       "            3.6208e-01, -9.1427e-01],\n",
       "          [ 9.0224e-01,  1.9611e-01, -5.0571e-01,  ..., -5.4598e-01,\n",
       "            3.2713e-01, -6.5325e-01]],\n",
       "\n",
       "         [[-1.0899e+00,  1.2574e+00, -1.4994e+00,  ..., -1.0651e-01,\n",
       "           -2.9514e-01,  2.1083e-02],\n",
       "          [-5.1164e-01,  1.1904e+00, -7.1602e-01,  ...,  1.4060e-01,\n",
       "           -4.6783e-01,  2.0951e-01],\n",
       "          [-4.1726e-01,  1.2079e+00, -4.4966e-01,  ...,  1.2981e-01,\n",
       "           -4.7805e-01, -9.9640e-02],\n",
       "          ...,\n",
       "          [-1.2199e+00,  1.1540e+00, -2.1711e-01,  ..., -2.3492e-01,\n",
       "           -5.5538e-01, -7.2347e-02],\n",
       "          [-1.0995e+00,  7.0646e-01, -6.7959e-01,  ..., -6.5123e-01,\n",
       "           -1.1216e+00, -8.1717e-02],\n",
       "          [-1.3929e+00,  2.0645e+00,  6.1140e-01,  ..., -3.1992e-01,\n",
       "           -1.5649e+00,  4.6385e-01]],\n",
       "\n",
       "         [[ 1.1277e-01, -4.0483e-01,  5.3092e-01,  ..., -4.7810e-01,\n",
       "           -4.8690e-01, -1.7816e-01],\n",
       "          [-4.0800e-03,  3.7986e-01,  6.6523e-02,  ..., -7.1946e-01,\n",
       "           -7.4477e-02, -4.2794e-01],\n",
       "          [ 5.9232e-02,  3.6244e-01,  2.7822e-01,  ..., -6.9334e-01,\n",
       "           -6.8792e-01, -1.2543e-01],\n",
       "          ...,\n",
       "          [ 2.3271e-01,  1.5773e-01,  4.9331e-04,  ...,  1.9113e-02,\n",
       "           -4.1970e-01,  2.3357e-01],\n",
       "          [ 2.4139e-01, -6.8256e-02,  3.2153e-01,  ..., -2.6026e-01,\n",
       "           -4.9188e-01,  4.0271e-01],\n",
       "          [ 1.1072e-02,  4.2004e-01,  3.7807e-01,  ..., -1.2854e+00,\n",
       "            3.9922e-01,  6.9830e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.7588e-01, -1.2803e+00,  8.3935e-02,  ...,  7.4655e-01,\n",
       "           -6.3536e-03, -1.0844e+00],\n",
       "          [ 9.6815e-02, -3.9877e-01,  3.5982e-01,  ..., -2.3760e-02,\n",
       "            1.2827e-02, -6.2986e-01],\n",
       "          [ 4.3357e-01, -1.2273e+00, -1.9412e-01,  ...,  3.9151e-01,\n",
       "           -4.6767e-01, -7.4091e-01],\n",
       "          ...,\n",
       "          [ 1.8486e-01,  1.4683e-02,  5.2612e-01,  ...,  5.6017e-01,\n",
       "            2.4871e-01, -6.5352e-01],\n",
       "          [ 4.8528e-01, -1.3589e+00, -2.5664e-01,  ...,  5.3549e-01,\n",
       "           -7.5459e-02, -9.2556e-01],\n",
       "          [-2.7672e-01, -4.5645e-01, -4.6709e-01,  ...,  8.2643e-01,\n",
       "            1.0190e+00, -2.2712e-01]],\n",
       "\n",
       "         [[ 1.1562e+00, -7.0038e-01,  6.1206e-01,  ...,  5.1271e-01,\n",
       "            3.3029e-01,  8.0170e-01],\n",
       "          [ 2.6690e-01, -5.6516e-01,  1.9995e-01,  ..., -7.3597e-01,\n",
       "            6.0135e-01, -1.2394e-01],\n",
       "          [ 5.5343e-01, -5.0621e-01,  6.3734e-01,  ..., -6.0615e-01,\n",
       "            5.9105e-01,  3.6875e-01],\n",
       "          ...,\n",
       "          [ 6.0255e-01, -6.3391e-01, -7.7835e-02,  ..., -3.2896e-01,\n",
       "            1.1865e+00,  5.8242e-01],\n",
       "          [ 9.0005e-01, -5.1592e-01,  7.5542e-01,  ...,  1.3251e-02,\n",
       "            8.0116e-01,  1.0300e+00],\n",
       "          [-3.1039e-02,  4.7649e-01,  2.4156e-01,  ..., -3.1489e-01,\n",
       "            1.0783e+00, -1.9197e-01]],\n",
       "\n",
       "         [[-9.9465e-02,  1.1558e-01,  5.6546e-01,  ..., -7.3387e-01,\n",
       "            3.6267e-01, -1.8555e-03],\n",
       "          [ 1.6058e-01, -1.0974e-01,  2.9730e-01,  ..., -4.4324e-01,\n",
       "           -3.5169e-01, -6.4800e-01],\n",
       "          [-2.6973e-01, -2.2991e-01,  2.2125e-01,  ..., -5.5574e-01,\n",
       "            3.2459e-01,  1.2874e-01],\n",
       "          ...,\n",
       "          [-7.1471e-02, -8.3791e-01, -1.9387e-01,  ..., -5.0620e-01,\n",
       "           -4.4683e-01, -9.3246e-03],\n",
       "          [-3.1139e-01, -6.2366e-01,  4.2198e-01,  ..., -8.7548e-01,\n",
       "            3.8605e-01, -1.2274e-01],\n",
       "          [-2.4210e-01, -2.7787e-01, -2.0921e-01,  ..., -1.3765e-02,\n",
       "           -3.0903e-01,  3.3705e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-3.9762e-01,  1.6887e-01, -1.4201e+00,  ...,  2.7147e-01,\n",
       "            1.3254e-01,  9.5813e-01],\n",
       "          [ 3.0216e-01, -8.3041e-01, -7.3475e-01,  ...,  3.2361e-01,\n",
       "            4.0187e-01,  1.2422e-01],\n",
       "          [ 4.8257e-02, -2.3896e-01, -6.0158e-01,  ...,  1.0673e-01,\n",
       "           -4.8623e-02, -2.7774e-02],\n",
       "          ...,\n",
       "          [ 4.3092e-01, -6.2134e-03,  1.0316e-01,  ..., -4.9281e-01,\n",
       "            2.6794e-01,  6.8663e-01],\n",
       "          [-1.9820e-01,  2.1346e-01, -7.9452e-01,  ..., -6.8198e-01,\n",
       "           -4.1971e-01,  7.2072e-01],\n",
       "          [ 2.2227e-01,  8.6449e-02, -1.5883e-01,  ...,  5.7475e-01,\n",
       "            9.5683e-01,  1.1927e+00]],\n",
       "\n",
       "         [[ 1.1434e+00, -4.7985e-01, -2.2152e-01,  ..., -1.6912e+00,\n",
       "           -1.5314e-01, -7.0825e-02],\n",
       "          [ 6.7454e-01, -4.0057e-02, -3.8708e-02,  ...,  8.3181e-02,\n",
       "           -2.8820e-01,  1.5155e-01],\n",
       "          [ 1.0971e+00, -7.5341e-02, -1.0909e-01,  ..., -5.1887e-01,\n",
       "           -1.4698e-01, -1.5478e-01],\n",
       "          ...,\n",
       "          [ 1.2145e-01, -1.6778e-02, -4.9347e-01,  ..., -2.2558e-01,\n",
       "            3.1084e-01, -6.0871e-01],\n",
       "          [ 1.1841e-01, -3.0124e-01, -3.3189e-01,  ..., -7.7096e-01,\n",
       "           -1.9877e-01, -2.5925e-01],\n",
       "          [-6.2495e-01, -1.1223e+00,  1.2952e-01,  ..., -9.4636e-01,\n",
       "           -3.6144e-01, -5.1282e-01]],\n",
       "\n",
       "         [[-2.3943e-01, -2.7483e-01, -2.2521e-01,  ...,  1.1586e-01,\n",
       "            7.1174e-01,  1.7211e-01],\n",
       "          [-2.1530e-01, -4.3745e-03,  6.9482e-02,  ...,  7.5040e-02,\n",
       "            7.1238e-01,  1.5672e-01],\n",
       "          [-3.9533e-01, -5.7173e-01, -4.5401e-02,  ..., -1.2205e-01,\n",
       "            7.4706e-01,  5.4194e-01],\n",
       "          ...,\n",
       "          [-9.5108e-01, -1.3456e-01, -1.2308e-02,  ..., -7.6075e-02,\n",
       "            7.6883e-01,  5.2965e-01],\n",
       "          [-6.8589e-01,  4.4216e-02,  2.0534e-01,  ..., -4.4045e-01,\n",
       "            1.0349e+00,  6.4611e-02],\n",
       "          [-1.1629e+00, -1.4282e+00, -8.9854e-03,  ...,  1.4315e-01,\n",
       "            1.1860e+00,  9.9220e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.2606e+00, -7.0408e-01,  7.8363e-02,  ...,  1.6784e-01,\n",
       "           -9.0276e-02, -2.9782e-02],\n",
       "          [-1.0648e+00,  2.4537e-01, -2.9420e-01,  ...,  2.0623e-01,\n",
       "           -4.0356e-01,  3.6760e-01],\n",
       "          [-1.1450e+00, -4.3609e-02, -5.8765e-01,  ...,  1.3007e-01,\n",
       "           -6.4011e-01,  1.8106e-01],\n",
       "          ...,\n",
       "          [-3.1704e-01,  2.3962e-01, -2.6911e-01,  ...,  1.8404e-01,\n",
       "           -7.6147e-02,  3.5297e-01],\n",
       "          [-1.1850e+00, -3.7684e-01, -4.9489e-01,  ..., -1.4437e-01,\n",
       "           -8.5172e-03,  1.7684e-01],\n",
       "          [-6.4454e-01,  9.2689e-02,  5.3042e-01,  ...,  3.4677e-01,\n",
       "           -8.0817e-02, -1.4781e+00]],\n",
       "\n",
       "         [[ 1.7528e-02,  3.5249e-01,  3.9031e-01,  ..., -1.8651e-01,\n",
       "           -5.0850e-01, -2.1084e-01],\n",
       "          [ 1.4104e-01,  8.0685e-02, -4.5084e-01,  ..., -1.2543e-01,\n",
       "           -2.5006e-02,  3.9391e-01],\n",
       "          [ 1.2070e-01,  3.9641e-01, -2.3352e-01,  ...,  3.1424e-01,\n",
       "           -7.3265e-02,  4.5537e-01],\n",
       "          ...,\n",
       "          [-4.9210e-03,  9.5131e-02, -5.0424e-01,  ...,  4.8649e-01,\n",
       "            3.3027e-01,  6.3313e-01],\n",
       "          [-2.1282e-01,  2.8190e-01,  1.1264e-01,  ..., -4.1163e-01,\n",
       "           -1.4307e-01, -1.9866e-01],\n",
       "          [-8.6139e-03,  1.2363e+00, -6.8657e-01,  ...,  4.2705e-01,\n",
       "           -8.6051e-01,  5.9522e-01]],\n",
       "\n",
       "         [[-7.1492e-01,  1.1491e-01,  5.9761e-01,  ..., -2.6771e-02,\n",
       "            1.0978e-01, -1.0217e-01],\n",
       "          [-6.7664e-01, -1.8198e-01,  4.9648e-01,  ..., -3.0207e-01,\n",
       "            4.2341e-01, -4.8698e-01],\n",
       "          [-6.1157e-01, -1.0569e-01,  9.1813e-01,  ...,  2.7843e-01,\n",
       "           -1.3091e-01, -3.6423e-01],\n",
       "          ...,\n",
       "          [-1.8135e-01,  7.3640e-01,  7.6328e-01,  ...,  3.3300e-01,\n",
       "           -8.8685e-02, -1.1722e+00],\n",
       "          [-1.2682e-02,  1.5975e-01,  2.2989e-01,  ..., -7.0598e-02,\n",
       "            1.9320e-02, -7.8179e-01],\n",
       "          [-1.2984e-01,  1.0758e+00,  5.3060e-01,  ...,  6.7016e-01,\n",
       "            3.4727e-01, -5.1732e-01]]],\n",
       "\n",
       "\n",
       "        [[[-3.7047e-01,  1.4804e-01, -1.3839e+00,  ...,  2.4997e-01,\n",
       "            1.0302e-01,  9.8248e-01],\n",
       "          [ 6.0784e-01, -3.7318e-01, -9.8888e-01,  ...,  3.7820e-01,\n",
       "           -3.9264e-01,  5.5345e-01],\n",
       "          [ 7.9082e-02, -2.6273e-01, -6.4130e-01,  ..., -1.1951e-01,\n",
       "            3.2720e-01,  2.7567e-01],\n",
       "          ...,\n",
       "          [ 6.0914e-01, -1.3077e-02, -7.5150e-01,  ..., -6.7580e-01,\n",
       "            9.1493e-01,  5.7765e-01],\n",
       "          [-1.9095e-01, -1.7149e-01, -5.8788e-01,  ..., -2.0758e-01,\n",
       "            7.2262e-01,  1.2187e+00],\n",
       "          [-5.9637e-01,  6.7064e-01, -4.9031e-01,  ...,  1.6539e-01,\n",
       "            4.6772e-01,  1.2003e+00]],\n",
       "\n",
       "         [[ 1.1385e+00, -5.0177e-01, -2.2474e-01,  ..., -1.6817e+00,\n",
       "           -1.1940e-01, -5.8742e-02],\n",
       "          [ 2.7400e-01, -2.2720e-01, -4.9617e-01,  ..., -3.2928e-02,\n",
       "           -1.2225e-01,  4.7730e-01],\n",
       "          [ 8.0113e-01, -8.7111e-02, -8.4080e-02,  ..., -2.3260e-01,\n",
       "           -3.1944e-01, -2.8652e-01],\n",
       "          ...,\n",
       "          [ 2.5435e-01, -2.7268e-01, -2.7149e-01,  ..., -3.3714e-01,\n",
       "            3.5684e-01,  1.7339e-01],\n",
       "          [ 8.3452e-01, -5.3734e-01, -1.2735e-03,  ..., -8.3957e-01,\n",
       "            1.2582e-01, -1.3629e-01],\n",
       "          [ 9.3784e-02, -7.2172e-01, -3.5120e-01,  ..., -1.1180e+00,\n",
       "            9.2906e-02,  2.7249e-02]],\n",
       "\n",
       "         [[-2.3069e-01, -2.6692e-01, -1.9869e-01,  ...,  1.4647e-01,\n",
       "            7.0371e-01,  1.8438e-01],\n",
       "          [-9.1735e-01, -8.9317e-01, -3.4793e-01,  ..., -4.8409e-01,\n",
       "            1.0145e+00,  1.8232e-01],\n",
       "          [-8.5739e-01, -8.3036e-01,  4.9158e-01,  ..., -2.7822e-01,\n",
       "            9.7995e-01,  4.2170e-01],\n",
       "          ...,\n",
       "          [-8.5988e-01, -1.4029e-01, -7.5985e-02,  ..., -2.4758e-01,\n",
       "            1.0514e+00,  5.7487e-01],\n",
       "          [-8.8774e-01, -1.1962e-01,  3.2344e-01,  ..., -4.3238e-03,\n",
       "            1.3878e+00,  7.7254e-01],\n",
       "          [-1.3570e+00, -1.0556e+00,  9.5422e-02,  ...,  5.0009e-01,\n",
       "            1.3835e+00,  6.4122e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.2225e+00, -7.3976e-01,  1.0610e-01,  ...,  1.5762e-01,\n",
       "           -1.0221e-01, -4.3487e-02],\n",
       "          [-6.9442e-01,  7.1912e-02, -2.4265e-01,  ...,  3.7568e-01,\n",
       "           -5.7452e-01,  1.5547e-01],\n",
       "          [-8.9469e-01, -1.1842e-01, -1.4322e-01,  ...,  1.7710e-01,\n",
       "           -4.0824e-01,  1.0503e-01],\n",
       "          ...,\n",
       "          [-5.4983e-01, -1.3217e-01, -1.5616e-01,  ...,  2.7199e-01,\n",
       "            2.7290e-01, -9.7931e-02],\n",
       "          [-6.7541e-01, -7.4235e-02,  1.6055e-01,  ..., -1.1911e-01,\n",
       "           -1.9379e-02,  1.9457e-01],\n",
       "          [-7.8171e-01, -9.4056e-02,  2.4642e-01,  ...,  1.6676e-02,\n",
       "           -3.5145e-02, -8.0046e-01]],\n",
       "\n",
       "         [[ 3.7493e-02,  3.1037e-01,  4.3715e-01,  ..., -2.0519e-01,\n",
       "           -5.0150e-01, -2.3931e-01],\n",
       "          [ 4.7822e-03,  1.9523e-01,  2.8330e-01,  ...,  1.8963e-01,\n",
       "           -7.0091e-01,  2.7311e-01],\n",
       "          [ 1.0175e-01,  1.9809e-01, -3.0315e-01,  ..., -4.8778e-02,\n",
       "           -2.1664e-01, -5.2875e-02],\n",
       "          ...,\n",
       "          [ 2.1413e-01, -9.8039e-02,  1.1262e-01,  ...,  3.4377e-01,\n",
       "           -4.4301e-01, -1.4096e-02],\n",
       "          [ 2.5447e-01, -1.5558e-01, -5.0816e-03,  ...,  3.2501e-01,\n",
       "            6.5165e-02,  3.2750e-01],\n",
       "          [ 2.0707e-01,  5.0221e-01, -2.2661e-01,  ..., -1.4792e-01,\n",
       "           -7.1104e-01,  3.4004e-01]],\n",
       "\n",
       "         [[-7.1980e-01,  6.9565e-02,  6.0378e-01,  ...,  4.0879e-03,\n",
       "            1.0899e-01, -9.5554e-02],\n",
       "          [-9.1674e-03,  1.5331e-01,  2.6182e-01,  ..., -3.7169e-01,\n",
       "            2.9962e-01, -6.6941e-01],\n",
       "          [-3.6633e-01,  1.5112e-01,  1.3018e+00,  ..., -3.0456e-02,\n",
       "           -3.4833e-02, -7.8842e-01],\n",
       "          ...,\n",
       "          [-4.0527e-01,  5.1802e-01,  4.4909e-01,  ...,  4.9912e-01,\n",
       "           -3.9879e-02, -8.1484e-01],\n",
       "          [-4.1480e-01, -6.9024e-02,  1.1164e+00,  ...,  1.8369e-01,\n",
       "           -4.7500e-01, -9.6472e-02],\n",
       "          [-6.8224e-02,  6.4799e-01,  5.6999e-01,  ...,  7.3760e-01,\n",
       "           -1.6439e-02, -4.0490e-02]]],\n",
       "\n",
       "\n",
       "        [[[-4.0145e-01,  1.7071e-01, -1.3983e+00,  ...,  2.6383e-01,\n",
       "            1.1844e-01,  9.7386e-01],\n",
       "          [-5.2916e-01,  7.3169e-01, -9.2616e-01,  ..., -1.0219e-01,\n",
       "            2.1590e-01,  1.0440e+00],\n",
       "          [-5.6312e-01,  5.7220e-01, -1.3336e+00,  ...,  3.7616e-01,\n",
       "            3.6759e-01,  1.3499e+00],\n",
       "          ...,\n",
       "          [-2.9319e-01,  2.9656e-01, -6.0157e-01,  ...,  1.7477e-01,\n",
       "            4.6756e-01,  5.7934e-01],\n",
       "          [-4.0658e-01,  3.6447e-01, -1.1822e+00,  ..., -4.9172e-02,\n",
       "           -1.4579e-01,  8.6461e-01],\n",
       "          [-4.7262e-01,  2.9573e-01, -7.9140e-02,  ...,  4.0200e-02,\n",
       "            1.0563e+00,  5.9207e-01]],\n",
       "\n",
       "         [[ 1.1394e+00, -4.8137e-01, -2.3128e-01,  ..., -1.6695e+00,\n",
       "           -1.2827e-01, -8.3848e-02],\n",
       "          [ 6.5195e-01, -3.3250e-01, -2.4331e-01,  ..., -1.0201e+00,\n",
       "           -2.4699e-01,  1.5888e-01],\n",
       "          [ 1.0196e+00, -4.6082e-01, -1.5590e-01,  ..., -1.4131e+00,\n",
       "           -9.4055e-02, -1.4719e-01],\n",
       "          ...,\n",
       "          [ 1.4672e-02, -7.8930e-02,  1.8428e-01,  ..., -2.8721e-01,\n",
       "           -6.6578e-01, -5.6653e-01],\n",
       "          [ 4.2850e-01, -4.1695e-01, -6.1019e-01,  ..., -9.3833e-01,\n",
       "           -2.8814e-01,  1.9490e-02],\n",
       "          [ 2.0639e-02, -4.9746e-01, -5.0298e-01,  ..., -8.0917e-01,\n",
       "           -3.7584e-01, -5.3134e-01]],\n",
       "\n",
       "         [[-2.2907e-01, -2.9979e-01, -2.1828e-01,  ...,  1.3680e-01,\n",
       "            7.3360e-01,  1.5116e-01],\n",
       "          [-4.2553e-01,  2.3883e-01, -8.5160e-02,  ..., -1.4000e-01,\n",
       "            5.8662e-01,  3.5533e-01],\n",
       "          [-5.3336e-01,  4.0602e-02, -9.0693e-02,  ...,  1.7057e-01,\n",
       "            9.0040e-01,  3.1820e-01],\n",
       "          ...,\n",
       "          [-6.7548e-01, -3.4240e-01, -2.9026e-01,  ...,  1.4927e-01,\n",
       "            1.0606e+00,  3.3230e-02],\n",
       "          [-7.3049e-01, -3.6294e-01, -2.5189e-01,  ...,  1.0280e-01,\n",
       "            1.0902e+00,  3.8782e-01],\n",
       "          [-1.6511e+00, -6.9024e-01,  1.6720e-01,  ...,  3.2882e-01,\n",
       "            1.3916e+00,  5.8314e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.2297e+00, -7.3889e-01,  9.4289e-02,  ...,  1.6513e-01,\n",
       "           -1.0186e-01, -4.4633e-02],\n",
       "          [-7.0237e-01, -4.3293e-01, -9.4813e-02,  ..., -1.1804e-01,\n",
       "            3.0118e-01,  2.5815e-01],\n",
       "          [-8.9534e-01, -9.4680e-01,  7.8244e-02,  ..., -5.3359e-01,\n",
       "            9.0849e-03,  8.7612e-02],\n",
       "          ...,\n",
       "          [-6.1353e-01, -2.3129e-01,  3.3022e-01,  ...,  1.6482e-01,\n",
       "            2.7521e-01,  5.2840e-01],\n",
       "          [-1.7076e+00, -4.3673e-01,  8.3652e-02,  ..., -5.7907e-01,\n",
       "            1.7402e-01, -1.6284e-01],\n",
       "          [-7.7395e-01, -2.7332e-01,  6.7123e-01,  ...,  1.7663e-01,\n",
       "            3.5330e-01, -1.3967e+00]],\n",
       "\n",
       "         [[ 4.8973e-02,  3.1276e-01,  4.0858e-01,  ..., -1.8792e-01,\n",
       "           -4.9213e-01, -2.2976e-01],\n",
       "          [ 2.8654e-01, -1.6161e-01, -2.6722e-02,  ..., -4.5076e-01,\n",
       "           -9.4576e-02, -1.0185e-02],\n",
       "          [ 6.3990e-03, -1.0999e-01, -2.8605e-01,  ..., -2.7971e-01,\n",
       "           -4.5621e-01, -4.5695e-01],\n",
       "          ...,\n",
       "          [-1.6490e-01, -1.8770e-01, -7.7367e-01,  ..., -1.4555e-01,\n",
       "            6.7003e-02,  2.8450e-02],\n",
       "          [ 1.4505e-01,  4.3084e-02, -5.1275e-01,  ...,  1.8282e-01,\n",
       "           -3.0360e-02, -1.4779e-01],\n",
       "          [ 5.9766e-02,  5.9636e-01, -9.3812e-01,  ..., -1.5794e-01,\n",
       "           -1.0489e+00,  4.8376e-01]],\n",
       "\n",
       "         [[-7.1954e-01,  1.0033e-01,  5.8879e-01,  ...,  1.4364e-02,\n",
       "            1.0772e-01, -1.0033e-01],\n",
       "          [-1.6191e-01, -7.4728e-02,  5.9985e-01,  ..., -2.9252e-01,\n",
       "            3.5976e-01, -4.5032e-01],\n",
       "          [-6.8710e-01,  1.2688e-01,  6.9592e-01,  ..., -3.8488e-02,\n",
       "           -8.5929e-02,  5.1001e-02],\n",
       "          ...,\n",
       "          [-4.7724e-01,  2.2949e-01,  5.3786e-01,  ..., -1.1948e-01,\n",
       "            5.1574e-02, -4.5837e-01],\n",
       "          [-5.5692e-01,  4.0110e-01,  4.2538e-01,  ...,  3.4529e-01,\n",
       "           -3.3410e-01, -1.3271e-01],\n",
       "          [ 5.5458e-02,  7.0859e-01,  8.8418e-02,  ...,  6.4585e-01,\n",
       "            1.5394e-02, -1.6208e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.3300e+00,  4.8769e+00, -1.3019e+00,  ..., -1.7052e-01,\n",
       "           -2.0988e+00, -8.1405e-04],\n",
       "          [-9.7751e-02,  1.0702e+00,  4.1779e-01,  ..., -6.1983e-01,\n",
       "           -2.4215e-01,  8.4433e-01],\n",
       "          [-1.4031e+00,  2.1169e+00,  2.3117e+00,  ...,  3.2384e+00,\n",
       "           -1.9989e+00,  8.1232e-01],\n",
       "          ...,\n",
       "          [ 2.0319e+00, -8.6973e-01,  5.0364e-01,  ...,  2.5799e+00,\n",
       "            2.1362e+00,  2.6048e-01],\n",
       "          [-1.8143e+00,  9.0786e-01,  1.3746e+00,  ...,  8.6653e-01,\n",
       "           -1.1742e+00, -9.8676e-01],\n",
       "          [ 2.5566e+00,  1.8760e+00, -7.6754e-01,  ...,  4.3803e+00,\n",
       "           -2.8127e-02, -3.3779e+00]],\n",
       "\n",
       "         [[-6.8022e-01, -9.4895e-01, -2.3712e+00,  ...,  1.0801e+00,\n",
       "            1.7921e+00,  1.1830e+00],\n",
       "          [ 1.1974e+00, -1.3164e+00, -6.4009e+00,  ...,  1.9558e+00,\n",
       "            2.4148e+00,  4.0170e-01],\n",
       "          [ 1.6826e+00, -2.7279e+00, -2.6911e+00,  ...,  5.4609e-01,\n",
       "           -1.6105e-02, -1.2072e-01],\n",
       "          ...,\n",
       "          [-3.5216e+00, -6.7411e-01, -2.4137e+00,  ...,  1.5297e+00,\n",
       "           -1.1631e+00,  3.1351e+00],\n",
       "          [-2.2076e+00, -2.2872e+00, -3.0042e+00,  ..., -5.6776e-01,\n",
       "           -2.0525e+00, -2.3914e+00],\n",
       "          [ 2.7636e-01, -3.2836e+00,  3.7855e+00,  ..., -1.3585e+00,\n",
       "            2.0900e+00,  2.1191e+00]],\n",
       "\n",
       "         [[-1.9546e+00, -1.7136e+00, -7.3546e-01,  ..., -2.5080e+00,\n",
       "            2.9600e+00, -1.1666e-01],\n",
       "          [-2.9020e+00,  1.2356e+00, -3.2129e+00,  ...,  1.7467e-01,\n",
       "            1.0273e+00, -6.4498e-01],\n",
       "          [-5.7472e-01, -9.6451e-01, -2.2310e+00,  ...,  2.4318e+00,\n",
       "            1.7332e+00, -3.1341e+00],\n",
       "          ...,\n",
       "          [-3.8552e+00, -3.9994e-01, -1.9831e+00,  ...,  2.8602e-01,\n",
       "           -2.9315e+00, -3.1278e+00],\n",
       "          [ 2.9238e-01, -2.4281e+00, -3.5898e+00,  ...,  3.9839e-01,\n",
       "           -2.3930e+00, -6.4754e-01],\n",
       "          [ 1.1154e+00, -1.8740e+00,  2.1671e+00,  ..., -2.6136e+00,\n",
       "            4.6853e+00, -1.8621e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.4290e+00,  6.0557e-01, -5.9677e-01,  ..., -6.7900e-02,\n",
       "           -1.3163e-01,  9.1452e-02],\n",
       "          [-2.0130e-01,  5.1905e-01,  1.6510e+00,  ..., -1.8736e+00,\n",
       "           -6.5182e-01,  1.3288e+00],\n",
       "          [ 2.4024e+00,  6.7137e-01,  1.0227e+00,  ..., -2.4750e-01,\n",
       "            6.7479e-01, -9.3281e-05],\n",
       "          ...,\n",
       "          [ 1.3023e+00, -3.9106e+00,  2.5569e-02,  ..., -2.8894e+00,\n",
       "           -2.9842e-01,  6.8919e+00],\n",
       "          [ 1.6003e+00, -1.8793e+00,  2.4644e-01,  ...,  7.1097e-01,\n",
       "           -2.5701e+00,  1.4277e+00],\n",
       "          [-1.2311e+00,  7.0697e-01, -4.6663e+00,  ...,  1.5001e+00,\n",
       "           -2.0031e+00, -1.0101e+00]],\n",
       "\n",
       "         [[ 9.9498e-01, -2.3986e+00,  2.3455e+00,  ...,  2.0704e+00,\n",
       "           -8.5965e-02, -7.8335e-01],\n",
       "          [ 1.3476e+00, -1.9229e+00,  1.3246e+00,  ...,  2.2128e+00,\n",
       "            1.1079e+00,  1.5623e+00],\n",
       "          [-5.2859e-01,  5.9649e-01,  2.0985e+00,  ...,  1.3778e+00,\n",
       "           -1.8629e+00,  5.0898e+00],\n",
       "          ...,\n",
       "          [ 8.7209e-01, -3.3189e+00, -1.2868e+00,  ..., -1.0008e-01,\n",
       "            4.1332e-01,  3.5973e+00],\n",
       "          [ 1.9271e-01,  5.1718e-01,  1.0627e+00,  ...,  4.8143e-01,\n",
       "            1.2493e+00,  3.1110e+00],\n",
       "          [-2.5622e+00, -4.5106e-01,  7.9149e-01,  ..., -1.4096e+00,\n",
       "           -1.3621e-02, -1.1065e+00]],\n",
       "\n",
       "         [[-1.5339e+00, -7.4992e-01, -1.5824e+00,  ..., -7.7870e-01,\n",
       "            1.8227e+00, -4.2133e-02],\n",
       "          [ 1.8469e+00, -1.9474e+00, -1.5575e+00,  ...,  1.8345e+00,\n",
       "           -2.9421e-01, -1.5724e-01],\n",
       "          [ 1.2824e+00,  3.2806e-01, -2.0596e+00,  ..., -2.7613e-01,\n",
       "            7.1404e-02,  1.7913e+00],\n",
       "          ...,\n",
       "          [ 3.8525e+00,  1.5971e+00,  4.4587e-01,  ...,  1.2773e+00,\n",
       "            1.4557e+00, -3.0321e+00],\n",
       "          [ 1.7222e+00,  3.2001e+00, -8.4964e-01,  ...,  1.1309e+00,\n",
       "            7.5879e-01, -2.1363e+00],\n",
       "          [-2.7762e+00, -1.7842e+00, -3.4743e+00,  ..., -2.4471e+00,\n",
       "            1.8295e+00,  1.1593e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 1.0930e+00,  4.6885e-01, -2.4203e+00,  ...,  2.7861e+00,\n",
       "            2.2997e+00, -1.0815e+00],\n",
       "          [ 1.4155e-01,  5.8425e-01,  3.2314e+00,  ...,  4.4214e+00,\n",
       "           -1.2735e+00,  5.8478e-01],\n",
       "          [-7.7142e-01, -3.7401e-01, -4.3150e-02,  ...,  2.6774e+00,\n",
       "           -3.8735e+00, -2.1346e-02],\n",
       "          ...,\n",
       "          [-1.7792e+00,  8.5182e-01,  1.6028e+00,  ..., -1.1414e+00,\n",
       "           -9.6943e-01,  3.4126e-01],\n",
       "          [-1.9491e+00,  2.3414e+00,  2.5994e+00,  ..., -1.4592e+00,\n",
       "           -1.4985e-01,  4.3085e-01],\n",
       "          [ 2.5712e+00,  2.0111e+00, -7.1233e-01,  ...,  4.3633e+00,\n",
       "           -8.8785e-03, -3.3729e+00]],\n",
       "\n",
       "         [[ 1.1743e+00, -5.8916e-01,  3.4794e-01,  ...,  1.5656e+00,\n",
       "            3.0151e-01,  4.6155e+00],\n",
       "          [ 6.3063e-01, -1.7835e+00, -2.7130e+00,  ...,  4.0782e-01,\n",
       "           -1.4473e+00,  1.0396e+00],\n",
       "          [-1.7193e+00, -3.4651e-01, -2.3753e+00,  ...,  3.9884e+00,\n",
       "           -5.8792e-01,  1.3664e-01],\n",
       "          ...,\n",
       "          [-1.5877e+00,  1.1061e+00, -1.5699e+00,  ...,  3.6163e+00,\n",
       "           -2.2636e+00, -4.7611e-01],\n",
       "          [-4.3691e-01,  5.2735e-01, -1.1582e+00,  ..., -1.6220e+00,\n",
       "           -1.7937e+00,  7.0564e-01],\n",
       "          [ 3.0017e-01, -3.3410e+00,  3.7663e+00,  ..., -1.3219e+00,\n",
       "            2.1050e+00,  2.1437e+00]],\n",
       "\n",
       "         [[-1.8740e+00,  1.2873e+00,  1.0780e+00,  ...,  2.3261e+00,\n",
       "           -1.3834e+00, -2.8545e+00],\n",
       "          [-2.8263e-01, -7.1477e-01, -3.0854e+00,  ...,  1.8135e+00,\n",
       "            6.3921e-01, -3.3878e+00],\n",
       "          [ 6.5739e-01,  5.6779e-01, -1.1850e+00,  ...,  3.0425e+00,\n",
       "            9.2789e-01,  2.9382e+00],\n",
       "          ...,\n",
       "          [ 1.6457e+00, -1.5913e-01,  1.6756e+00,  ..., -2.9544e-02,\n",
       "           -1.5805e-01,  6.4888e-01],\n",
       "          [-1.4429e-02,  1.2971e+00, -1.2557e+00,  ..., -7.4654e-01,\n",
       "           -6.9073e-01, -2.2992e-01],\n",
       "          [ 1.0483e+00, -1.8457e+00,  2.2000e+00,  ..., -2.6460e+00,\n",
       "            4.7023e+00, -2.0164e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.0371e+00,  1.1579e+00, -4.4018e-01,  ...,  3.2814e-02,\n",
       "            6.3393e-02,  3.4140e-02],\n",
       "          [ 2.8909e+00,  7.5361e-01,  9.3715e-01,  ..., -1.1037e-01,\n",
       "           -7.7657e-01,  2.9156e+00],\n",
       "          [-1.9776e-01,  2.4389e+00,  1.0795e+00,  ...,  3.5794e-01,\n",
       "           -2.1351e+00, -4.2214e-01],\n",
       "          ...,\n",
       "          [ 7.9560e-01,  7.0813e-01,  4.1475e-01,  ..., -3.7473e-01,\n",
       "           -1.4000e+00,  1.7085e+00],\n",
       "          [-1.9440e-01, -2.4851e+00,  7.2197e-03,  ...,  1.1980e+00,\n",
       "           -1.8065e+00,  8.9763e-01],\n",
       "          [-1.2826e+00,  6.9739e-01, -4.6903e+00,  ...,  1.4987e+00,\n",
       "           -2.0899e+00, -9.3014e-01]],\n",
       "\n",
       "         [[ 3.7221e-01, -2.9846e+00, -1.9961e+00,  ...,  1.8683e+00,\n",
       "            2.9368e-03,  1.7259e+00],\n",
       "          [-4.0642e-01, -3.7719e-01,  6.9008e-01,  ...,  1.7769e-01,\n",
       "           -3.0362e+00,  4.9319e+00],\n",
       "          [ 1.7222e+00,  1.4841e+00,  2.5316e+00,  ..., -8.9828e-01,\n",
       "           -4.0937e-01,  2.8736e+00],\n",
       "          ...,\n",
       "          [ 1.6224e-01,  1.6718e+00,  3.4154e+00,  ..., -2.6958e-01,\n",
       "            3.5503e+00,  4.0708e-01],\n",
       "          [-4.9065e-01,  2.4281e+00,  1.4716e+00,  ...,  1.3905e+00,\n",
       "            4.7139e+00,  4.2072e-03],\n",
       "          [-2.5518e+00, -4.1185e-01,  8.0368e-01,  ..., -1.3004e+00,\n",
       "            4.3217e-02, -1.0706e+00]],\n",
       "\n",
       "         [[ 5.0323e-01, -6.6819e-01,  1.5383e+00,  ..., -2.7577e-01,\n",
       "            2.6189e+00,  3.0101e+00],\n",
       "          [ 1.0242e+00, -1.2900e-01, -2.3432e-01,  ...,  6.9437e-01,\n",
       "            9.2258e-01,  2.0663e+00],\n",
       "          [ 3.2154e-01,  4.1728e+00,  2.2986e-01,  ..., -6.3147e-02,\n",
       "            9.8811e-01,  1.6739e-01],\n",
       "          ...,\n",
       "          [-1.0442e+00,  3.2598e+00,  8.7378e-01,  ..., -2.5358e+00,\n",
       "           -4.9048e-01,  4.6979e-01],\n",
       "          [-1.9967e+00,  8.6837e-01, -8.2657e-01,  ..., -5.7828e-01,\n",
       "            6.0601e-01, -1.9246e+00],\n",
       "          [-2.8514e+00, -1.8751e+00, -3.4663e+00,  ..., -2.4850e+00,\n",
       "            1.8007e+00,  1.1929e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 1.2439e+00,  2.0731e+00, -4.6744e+00,  ..., -1.0658e-01,\n",
       "            1.0361e+00,  2.5586e-01],\n",
       "          [ 4.4398e-01,  2.7696e+00,  3.8283e-01,  ...,  3.3585e+00,\n",
       "           -2.6379e-01,  5.1775e-01],\n",
       "          [-1.4905e+00,  3.4560e-01,  9.3669e-01,  ..., -1.6233e+00,\n",
       "           -2.1330e+00, -9.2718e-01],\n",
       "          ...,\n",
       "          [-4.9004e-01,  5.2950e+00, -1.0960e-01,  ...,  1.4731e+00,\n",
       "            1.7452e-01, -4.8808e+00],\n",
       "          [-1.9848e+00,  6.9667e-02,  4.6454e-01,  ...,  1.3793e-01,\n",
       "           -6.6936e-01, -1.2760e+00],\n",
       "          [ 2.6124e+00,  1.9413e+00, -7.5879e-01,  ...,  4.3850e+00,\n",
       "           -7.3297e-02, -3.3794e+00]],\n",
       "\n",
       "         [[ 1.4381e+00,  1.3795e-02, -2.3552e+00,  ...,  1.0960e+00,\n",
       "           -1.7809e+00,  3.0145e+00],\n",
       "          [ 4.0926e+00, -2.6505e+00, -5.9680e-01,  ..., -3.4753e+00,\n",
       "           -1.3592e+00,  3.1457e+00],\n",
       "          [-5.1811e+00,  8.7885e-01,  1.1940e-01,  ..., -3.8571e+00,\n",
       "            2.3698e+00, -1.1474e-01],\n",
       "          ...,\n",
       "          [-2.1103e+00, -1.0333e+00, -2.4824e-01,  ..., -3.2900e+00,\n",
       "           -1.8886e+00, -3.1227e+00],\n",
       "          [-4.1419e+00, -8.7856e-01, -1.1032e+00,  ..., -2.3526e+00,\n",
       "           -3.6656e+00, -1.0249e+00],\n",
       "          [ 3.1261e-01, -3.2695e+00,  3.7793e+00,  ..., -1.4105e+00,\n",
       "            2.0854e+00,  2.1320e+00]],\n",
       "\n",
       "         [[-2.7444e+00,  6.4324e-01, -1.7537e+00,  ..., -1.6172e-01,\n",
       "           -1.6152e+00, -7.6766e-02],\n",
       "          [-1.0741e+00, -5.6179e-02, -7.8969e-01,  ..., -4.5888e-01,\n",
       "            5.3162e-01, -1.4861e+00],\n",
       "          [-3.0990e+00,  4.5695e-01,  2.1918e+00,  ..., -2.7690e+00,\n",
       "           -1.5578e+00, -1.1586e+00],\n",
       "          ...,\n",
       "          [-1.2718e+00, -1.4018e+00,  9.3495e-01,  ..., -1.9407e+00,\n",
       "           -2.0627e+00, -3.3464e+00],\n",
       "          [-1.6364e-01, -6.7069e-01, -1.6707e+00,  ..., -9.7989e-01,\n",
       "           -1.8600e+00, -6.1205e-01],\n",
       "          [ 1.0877e+00, -1.8526e+00,  2.1797e+00,  ..., -2.6592e+00,\n",
       "            4.7557e+00, -1.9667e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.5907e+00, -1.1189e-01, -1.1888e+00,  ..., -1.4370e-02,\n",
       "           -5.6544e-01,  1.4205e+00],\n",
       "          [ 3.1632e+00, -9.1938e-01, -3.1083e+00,  ..., -8.6366e-01,\n",
       "            1.9966e+00,  2.8269e+00],\n",
       "          [-1.0280e+00, -2.8986e+00, -7.8540e-01,  ..., -2.1771e+00,\n",
       "            1.9211e+00, -2.9439e-01],\n",
       "          ...,\n",
       "          [ 3.3773e-01, -2.5437e+00, -1.9491e-01,  ...,  6.9956e-01,\n",
       "           -1.2490e+00,  2.2166e-01],\n",
       "          [ 1.3232e+00, -1.2727e+00, -6.4628e-02,  ...,  2.6720e+00,\n",
       "            6.3293e-01,  5.6149e-01],\n",
       "          [-1.2479e+00,  6.3615e-01, -4.7159e+00,  ...,  1.5317e+00,\n",
       "           -2.0012e+00, -1.0209e+00]],\n",
       "\n",
       "         [[-1.4927e+00,  2.2138e-01, -6.6731e-01,  ...,  1.8047e+00,\n",
       "           -8.1958e-01,  4.2924e-01],\n",
       "          [-3.5420e+00, -7.5747e-01, -1.2049e+00,  ...,  5.1072e-01,\n",
       "           -3.4434e-01,  1.4156e+00],\n",
       "          [-3.8613e+00, -2.0769e+00,  2.0255e+00,  ...,  1.8754e+00,\n",
       "           -9.5256e-01,  7.3226e-01],\n",
       "          ...,\n",
       "          [-1.3370e+00,  2.2762e-01, -2.8928e-01,  ...,  2.5504e+00,\n",
       "           -1.7411e+00,  1.3396e+00],\n",
       "          [ 9.2344e-01, -9.9282e-01,  7.0511e-01,  ..., -2.7835e-01,\n",
       "            6.9870e-01,  3.7219e+00],\n",
       "          [-2.5822e+00, -3.6476e-01,  7.8468e-01,  ..., -1.3962e+00,\n",
       "            1.4526e-02, -1.0837e+00]],\n",
       "\n",
       "         [[-3.8612e-01, -1.0294e+00, -2.6528e+00,  ..., -1.9052e+00,\n",
       "           -1.7002e-01,  1.0750e+00],\n",
       "          [ 1.1283e+00, -1.7737e+00,  2.8246e-01,  ..., -3.1218e+00,\n",
       "            1.1342e+00, -6.1057e-01],\n",
       "          [-1.7700e+00,  2.3141e+00, -4.9271e+00,  ..., -4.3621e-01,\n",
       "           -2.8951e+00, -1.1481e+00],\n",
       "          ...,\n",
       "          [ 9.0899e-01,  1.7258e+00, -2.8039e+00,  ...,  3.4686e+00,\n",
       "           -7.2098e-02,  6.7170e-01],\n",
       "          [ 3.5472e-01,  2.0445e+00, -9.2276e-01,  ...,  3.5863e+00,\n",
       "            6.1567e-01, -5.4953e-01],\n",
       "          [-2.8256e+00, -1.8436e+00, -3.4876e+00,  ..., -2.3787e+00,\n",
       "            1.8023e+00,  1.2797e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-5.0946e-01,  1.2555e+00, -1.9718e-01,  ...,  1.7575e-01,\n",
       "            3.1410e-01, -3.8418e-01],\n",
       "          [ 5.5611e-01,  1.2681e+00,  1.4679e+00,  ...,  3.1648e-02,\n",
       "           -9.6114e-01, -2.8911e-01],\n",
       "          [ 6.8918e-01,  6.7753e-01, -5.3361e-01,  ...,  4.9985e-01,\n",
       "           -1.7269e+00, -1.4877e-01],\n",
       "          ...,\n",
       "          [ 1.7998e+00,  1.6436e+00, -1.5320e-01,  ...,  5.0522e-01,\n",
       "           -1.7890e+00, -3.6760e-01],\n",
       "          [-8.7244e-02,  1.5124e-01, -9.3404e-01,  ..., -5.2376e-01,\n",
       "           -1.0214e+00, -1.0735e-01],\n",
       "          [ 2.6872e-01,  3.5101e-02, -7.6156e-01,  ...,  8.7640e-02,\n",
       "            4.3284e-01, -9.9357e-02]],\n",
       "\n",
       "         [[ 2.1210e-01,  7.4209e-01, -1.7574e+00,  ...,  4.2441e-02,\n",
       "           -2.8787e-02, -2.0117e-01],\n",
       "          [-1.6282e+00, -7.6951e-01, -3.7274e-01,  ..., -2.8749e-01,\n",
       "            5.8865e-04, -1.5133e-01],\n",
       "          [-1.9243e+00, -1.3360e+00, -7.2441e-01,  ..., -2.2644e+00,\n",
       "           -2.1247e+00, -5.1560e-01],\n",
       "          ...,\n",
       "          [-1.4869e-01, -1.1508e-01, -1.1072e+00,  ..., -2.8171e+00,\n",
       "           -1.4591e-02,  1.3031e+00],\n",
       "          [-1.9977e-01,  2.0896e-01, -8.9695e-01,  ..., -1.3735e-01,\n",
       "           -7.0680e-01,  6.6272e-01],\n",
       "          [-8.4278e-02,  2.1943e-01, -4.3430e-01,  ...,  1.3998e-01,\n",
       "            2.0847e-01, -1.7475e-01]],\n",
       "\n",
       "         [[ 9.6517e-01, -1.9690e+00,  3.0457e-01,  ...,  1.1847e+00,\n",
       "            8.3354e-01, -8.1252e-01],\n",
       "          [ 4.0247e-01, -1.9532e+00,  7.6359e-01,  ...,  1.0620e+00,\n",
       "           -2.5962e-01,  8.8774e-01],\n",
       "          [-3.7834e-01, -5.1542e-01, -3.1593e-01,  ..., -1.8744e-01,\n",
       "            1.4310e-01,  1.7349e+00],\n",
       "          ...,\n",
       "          [-4.0910e-01, -1.4018e+00,  9.8942e-01,  ...,  9.5042e-01,\n",
       "           -7.7443e-01,  3.2707e-01],\n",
       "          [ 4.3691e-01, -1.9624e-01,  8.7414e-01,  ...,  3.9706e-02,\n",
       "            9.2942e-02, -7.4360e-01],\n",
       "          [ 3.0031e-01, -7.0928e-03,  3.8509e-01,  ..., -2.8236e-01,\n",
       "            1.8945e-01, -2.2535e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 6.2850e-01, -2.2469e-01, -4.0156e-01,  ..., -8.1004e-01,\n",
       "           -1.0921e+00, -1.5089e+00],\n",
       "          [-2.0264e+00,  4.6398e-01,  3.5184e-01,  ...,  1.9640e+00,\n",
       "            1.2748e-01,  5.1627e-01],\n",
       "          [-3.3988e+00, -2.1832e-01,  1.2317e+00,  ...,  2.7058e+00,\n",
       "            1.4605e-01,  2.8467e-01],\n",
       "          ...,\n",
       "          [-1.6901e+00, -2.2130e+00, -4.9368e-01,  ...,  1.8709e+00,\n",
       "           -2.2352e-01,  1.3407e+00],\n",
       "          [-6.4543e-01,  7.1880e-01, -1.1361e+00,  ...,  1.1660e+00,\n",
       "           -1.2672e+00, -1.4274e-01],\n",
       "          [ 3.8352e-02, -3.6770e-01,  2.1789e-01,  ...,  6.0162e-01,\n",
       "            2.8588e-01,  1.1901e-01]],\n",
       "\n",
       "         [[ 2.5690e-01, -2.7869e-01, -8.1283e-01,  ...,  1.3218e+00,\n",
       "            8.4067e-01, -4.3013e-02],\n",
       "          [ 1.8001e+00,  8.0169e-01,  4.5715e-01,  ...,  8.1478e-01,\n",
       "           -6.1282e-01,  2.1761e-01],\n",
       "          [ 1.2565e+00,  5.5028e-02, -4.2303e-01,  ..., -1.1702e+00,\n",
       "           -6.4511e-01,  2.3849e-01],\n",
       "          ...,\n",
       "          [ 6.8442e-01,  1.6093e+00, -4.8167e-01,  ..., -9.0541e-01,\n",
       "           -1.0459e+00,  3.2668e-01],\n",
       "          [ 1.8589e-02,  4.8651e-01, -3.7148e-02,  ..., -7.5288e-01,\n",
       "            9.5919e-01, -6.0025e-01],\n",
       "          [-3.5705e-01, -2.4186e-01,  5.0667e-01,  ...,  4.3438e-01,\n",
       "           -7.9176e-02, -7.5686e-01]],\n",
       "\n",
       "         [[-1.2271e+00,  1.0670e+00,  9.3724e-01,  ..., -6.2595e-02,\n",
       "           -5.2220e-01,  1.0638e+00],\n",
       "          [-1.5252e+00, -1.6297e+00,  1.3996e+00,  ...,  2.6565e-01,\n",
       "           -4.2613e-01, -8.0400e-01],\n",
       "          [ 2.3791e-01,  1.5922e-01,  3.0153e+00,  ..., -2.8253e-03,\n",
       "           -3.6880e-02,  6.2528e-01],\n",
       "          ...,\n",
       "          [-4.5819e-01, -8.7888e-01,  1.2481e+00,  ..., -1.6743e+00,\n",
       "           -4.8499e-01,  6.0143e-01],\n",
       "          [-1.0917e+00, -7.0596e-01,  2.7292e+00,  ..., -8.6258e-02,\n",
       "           -8.5958e-01,  1.9570e-01],\n",
       "          [-5.2503e-03,  8.9459e-02,  9.1978e-01,  ...,  5.5991e-02,\n",
       "            5.2968e-02,  3.2964e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 4.4855e-01,  1.1496e+00,  9.2363e-01,  ...,  1.2758e+00,\n",
       "            5.4430e-01, -8.9840e-01],\n",
       "          [ 1.9740e+00,  7.1769e-02, -1.0044e+00,  ...,  4.7536e-01,\n",
       "           -1.6123e+00, -7.8746e-01],\n",
       "          [ 2.3546e+00,  2.9038e-01, -9.3556e-01,  ...,  7.1974e-02,\n",
       "           -1.7799e+00, -1.4393e+00],\n",
       "          ...,\n",
       "          [-5.3520e-01,  1.6082e-01,  2.5046e-01,  ..., -1.2073e-01,\n",
       "           -9.3577e-01, -1.3469e-01],\n",
       "          [-2.4490e-02, -2.1915e-01, -6.9818e-01,  ..., -5.3345e-02,\n",
       "           -3.4275e-01,  3.3680e-01],\n",
       "          [ 2.8626e-01, -6.2756e-03, -7.3570e-01,  ...,  9.1583e-02,\n",
       "            4.3098e-01, -8.7147e-02]],\n",
       "\n",
       "         [[-8.0234e-01,  1.6187e+00,  4.9367e-01,  ...,  1.5675e+00,\n",
       "           -1.3470e+00,  1.7999e+00],\n",
       "          [-1.3107e+00,  4.8622e-01, -6.5571e-01,  ..., -2.3991e+00,\n",
       "           -1.5469e+00, -9.0145e-01],\n",
       "          [ 3.7436e-01,  2.5362e-01,  3.9172e-01,  ...,  8.0002e-01,\n",
       "           -3.0352e+00,  1.3091e+00],\n",
       "          ...,\n",
       "          [-4.9437e-01, -1.4068e+00, -3.8015e-01,  ..., -1.2558e+00,\n",
       "           -8.3955e-02, -3.6760e-01],\n",
       "          [ 4.9699e-01, -2.4722e-01,  5.6057e-02,  ..., -7.0458e-01,\n",
       "            7.4266e-01,  5.3001e-02],\n",
       "          [-1.0317e-01,  2.0208e-01, -4.3834e-01,  ...,  1.2214e-01,\n",
       "            2.1859e-01, -1.7524e-01]],\n",
       "\n",
       "         [[-2.7308e-01,  5.2995e-01, -6.0716e-01,  ..., -1.9581e-01,\n",
       "           -1.3251e+00,  3.1996e-01],\n",
       "          [ 2.1130e-01, -9.3667e-02, -1.5840e+00,  ..., -1.7924e-01,\n",
       "           -9.4852e-01,  1.1183e+00],\n",
       "          [-7.7682e-02, -2.3278e-02, -5.4433e-01,  ...,  1.7325e-01,\n",
       "           -2.6993e+00,  2.4465e-01],\n",
       "          ...,\n",
       "          [-1.5038e+00,  4.2749e-01, -2.8695e+00,  ..., -1.4576e+00,\n",
       "            8.8425e-01,  1.2834e+00],\n",
       "          [ 7.9209e-01,  3.3513e-01,  1.0627e+00,  ...,  9.0167e-01,\n",
       "            4.0000e-01, -7.9692e-01],\n",
       "          [ 3.1863e-01,  2.9625e-02,  3.8273e-01,  ..., -2.2286e-01,\n",
       "            2.2352e-01, -2.3350e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.7580e+00,  2.4487e-01, -5.2357e-01,  ...,  5.8409e-01,\n",
       "            1.3650e+00, -1.3861e+00],\n",
       "          [-2.1448e+00, -5.7704e-01,  9.3205e-01,  ...,  3.0628e+00,\n",
       "            9.7666e-02,  2.2407e-01],\n",
       "          [ 1.4074e+00, -1.0350e-01,  1.7056e+00,  ...,  2.9176e+00,\n",
       "            5.7503e-01,  2.2155e+00],\n",
       "          ...,\n",
       "          [-1.6131e+00,  5.6754e-01,  7.3481e-01,  ...,  1.6264e+00,\n",
       "           -2.1253e-01, -1.3689e+00],\n",
       "          [-3.3301e-01, -1.7704e-01, -3.7292e-01,  ...,  2.3007e-01,\n",
       "           -1.1835e+00, -3.6994e-01],\n",
       "          [-4.0218e-03, -3.8723e-01,  2.2892e-01,  ...,  6.0590e-01,\n",
       "            3.2760e-01,  1.1307e-01]],\n",
       "\n",
       "         [[-1.5812e+00,  2.4659e+00,  1.5479e+00,  ...,  1.1044e+00,\n",
       "            7.6329e-01, -5.3835e-02],\n",
       "          [ 1.1125e+00, -3.0243e-01, -3.4901e-01,  ..., -1.4188e+00,\n",
       "           -1.0020e+00,  4.3156e-02],\n",
       "          [ 1.8000e+00,  3.1248e-01,  7.4553e-01,  ...,  2.2725e+00,\n",
       "            7.0786e-01, -4.3944e-01],\n",
       "          ...,\n",
       "          [-1.0608e+00, -9.2093e-02,  3.1883e-01,  ...,  6.9244e-02,\n",
       "            9.8722e-01, -1.4307e+00],\n",
       "          [-5.1368e-01,  6.4373e-01,  1.3450e-01,  ...,  3.8797e-01,\n",
       "            1.1961e+00, -1.0911e+00],\n",
       "          [-3.7596e-01, -2.3357e-01,  5.0965e-01,  ...,  4.3819e-01,\n",
       "           -5.7160e-02, -7.8654e-01]],\n",
       "\n",
       "         [[ 7.4196e-02, -8.1358e-02, -7.1433e-01,  ..., -1.8361e+00,\n",
       "            1.3690e+00, -1.4303e+00],\n",
       "          [ 7.0909e-01, -2.8150e-01,  2.7615e+00,  ..., -5.5664e-01,\n",
       "           -7.9130e-02, -3.9220e-02],\n",
       "          [ 8.7227e-01, -1.6137e+00,  2.3103e+00,  ..., -4.2609e-01,\n",
       "           -3.9023e-01, -1.0250e+00],\n",
       "          ...,\n",
       "          [-3.8970e-01, -9.1551e-02,  1.6869e-01,  ..., -1.9519e+00,\n",
       "            2.8813e-01, -2.9167e-01],\n",
       "          [-8.9407e-02, -4.3226e-01,  1.7111e+00,  ..., -6.6257e-01,\n",
       "           -3.3568e-01,  1.4563e+00],\n",
       "          [ 1.1147e-02,  5.5733e-02,  9.1632e-01,  ...,  5.7107e-02,\n",
       "            3.5756e-02,  3.8148e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 2.9570e-01,  2.1572e+00, -7.1567e-01,  ..., -7.0767e-01,\n",
       "            2.8129e-02, -1.0639e-01],\n",
       "          [ 3.4474e-01,  1.6704e+00,  8.5055e-01,  ...,  9.3594e-01,\n",
       "           -7.4845e-01,  1.6804e+00],\n",
       "          [ 2.6541e-01,  8.7315e-01,  8.8734e-01,  ...,  1.3553e+00,\n",
       "            6.5329e-01,  6.5979e-01],\n",
       "          ...,\n",
       "          [ 1.2586e+00,  5.1584e-01,  5.1018e-01,  ...,  4.1876e-01,\n",
       "            3.4710e-02,  1.6621e+00],\n",
       "          [-4.1931e-01, -1.1147e-01, -1.1187e+00,  ..., -6.0366e-01,\n",
       "           -5.0082e-01, -6.1466e-01],\n",
       "          [ 2.4978e-01,  1.3001e-02, -7.3436e-01,  ...,  8.2134e-02,\n",
       "            4.5817e-01, -8.0562e-02]],\n",
       "\n",
       "         [[-6.4119e-01,  1.3199e+00, -6.6472e-01,  ..., -5.4467e-01,\n",
       "           -4.6156e-01, -1.1955e+00],\n",
       "          [ 1.8503e+00, -1.0240e+00,  5.6189e-01,  ..., -2.4360e+00,\n",
       "            7.1794e-01,  1.3652e-01],\n",
       "          [ 1.7822e-01,  5.9795e-01, -2.4458e+00,  ...,  5.5874e-01,\n",
       "            9.0299e-01,  2.3355e-01],\n",
       "          ...,\n",
       "          [ 3.3957e-01,  6.3194e-01,  8.0574e-02,  ..., -9.6331e-01,\n",
       "           -8.3099e-01,  8.8761e-01],\n",
       "          [ 5.9597e-01,  4.9010e-02, -8.5020e-02,  ..., -1.2795e-01,\n",
       "            8.2179e-01,  8.3164e-01],\n",
       "          [-6.4132e-02,  2.0481e-01, -3.9729e-01,  ...,  1.2749e-01,\n",
       "            2.3417e-01, -1.6733e-01]],\n",
       "\n",
       "         [[-7.8904e-01,  8.5128e-01,  9.7651e-01,  ..., -1.8867e+00,\n",
       "            9.1802e-01, -1.3348e+00],\n",
       "          [ 5.1012e-01, -1.5500e+00, -7.7471e-01,  ...,  5.4271e-01,\n",
       "           -6.9066e-01,  1.0809e+00],\n",
       "          [-1.3514e+00,  7.7765e-01, -1.2645e+00,  ...,  1.1825e+00,\n",
       "           -8.6429e-01,  6.2637e-01],\n",
       "          ...,\n",
       "          [-3.7847e-01, -8.1839e-01, -4.8605e-01,  ..., -8.1355e-02,\n",
       "           -7.4636e-01, -5.6465e-02],\n",
       "          [ 5.8517e-01, -6.0441e-01,  9.7197e-01,  ..., -7.3887e-01,\n",
       "           -1.2603e-01, -7.1912e-01],\n",
       "          [ 3.0556e-01,  3.2396e-02,  3.9525e-01,  ..., -2.8772e-01,\n",
       "            1.9599e-01, -2.6527e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-3.5916e+00, -1.4583e+00,  1.2756e+00,  ...,  8.5629e-01,\n",
       "           -2.8737e+00, -2.1571e+00],\n",
       "          [-1.7038e-01, -5.9842e-01,  6.0435e-01,  ...,  1.6409e+00,\n",
       "            9.9921e-01, -5.5447e-01],\n",
       "          [-7.1651e-01,  6.1352e-01, -6.2260e-01,  ...,  1.5300e+00,\n",
       "            8.7595e-01, -1.2955e+00],\n",
       "          ...,\n",
       "          [ 1.1997e+00, -1.5302e+00, -8.0774e-02,  ...,  1.3041e+00,\n",
       "            6.1706e-02,  1.8970e-01],\n",
       "          [-1.0891e-02,  2.0653e-01, -1.5299e+00,  ...,  1.1586e+00,\n",
       "           -1.0280e+00, -7.3448e-01],\n",
       "          [ 4.4441e-02, -3.6781e-01,  2.3404e-01,  ...,  5.9255e-01,\n",
       "            3.2338e-01,  1.3572e-01]],\n",
       "\n",
       "         [[ 1.6155e+00,  1.1373e+00, -9.5197e-01,  ...,  4.6142e-01,\n",
       "           -2.4773e+00, -5.3555e-01],\n",
       "          [ 3.2209e+00, -2.3379e-01, -3.5501e-01,  ...,  1.5940e+00,\n",
       "           -6.6055e-02, -1.6421e+00],\n",
       "          [-7.4082e-01, -4.5399e-01, -1.0181e+00,  ...,  2.8752e-01,\n",
       "           -1.7005e+00, -6.7240e-01],\n",
       "          ...,\n",
       "          [-2.8310e-01,  8.0070e-01, -2.6344e-01,  ...,  1.9748e-01,\n",
       "            7.3369e-01, -1.6897e-01],\n",
       "          [-2.0041e-01, -5.7008e-01, -1.0327e+00,  ..., -1.5598e-01,\n",
       "            5.1223e-01, -1.4062e+00],\n",
       "          [-4.0541e-01, -2.3993e-01,  5.4169e-01,  ...,  4.1730e-01,\n",
       "           -8.2126e-02, -7.7344e-01]],\n",
       "\n",
       "         [[-5.4442e-01, -1.2942e-01, -1.8708e-01,  ..., -9.5774e-01,\n",
       "            9.3902e-01,  2.2742e-01],\n",
       "          [-1.3556e+00,  7.6629e-01,  6.2979e-01,  ...,  1.3549e+00,\n",
       "            1.3266e+00, -7.8493e-01],\n",
       "          [-2.0700e+00, -1.3159e+00, -3.2740e-02,  ...,  1.7246e+00,\n",
       "            2.2589e-01, -5.9564e-01],\n",
       "          ...,\n",
       "          [ 2.6104e-01, -6.0430e-01,  1.8314e+00,  ..., -1.3384e+00,\n",
       "            1.2445e+00,  4.0039e-01],\n",
       "          [-6.8561e-01,  4.5321e-02,  1.7601e+00,  ..., -8.9750e-01,\n",
       "            4.7613e-01,  1.0473e+00],\n",
       "          [-7.6628e-04,  8.2722e-02,  8.9772e-01,  ...,  7.4442e-02,\n",
       "            8.9783e-02,  3.5612e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-1.4357e+00,  4.6405e-01, -9.1441e-01,  ...,  3.7303e-01,\n",
       "           -5.0882e-01, -8.1920e-01],\n",
       "          [-7.1019e-01,  2.0760e-01,  3.6359e-01,  ...,  1.6838e-01,\n",
       "           -5.0879e-01, -5.6860e-03],\n",
       "          [-3.6816e-01,  4.3424e-01,  1.4769e-01,  ...,  3.5422e-01,\n",
       "           -1.3911e-01,  1.0954e-01],\n",
       "          ...,\n",
       "          [-2.8940e-01, -9.2696e-02, -4.8299e-01,  ..., -1.1118e-01,\n",
       "           -9.8030e-01, -2.4886e-01],\n",
       "          [-4.2798e-01,  3.4090e-01, -1.6920e-01,  ..., -1.2139e-01,\n",
       "           -5.0380e-01, -2.2812e-01],\n",
       "          [-3.1389e-01,  4.1459e-01, -7.7436e-01,  ..., -1.0816e-02,\n",
       "           -1.2091e+00, -4.5292e-01]],\n",
       "\n",
       "         [[ 9.8530e-01,  1.5196e-01, -4.3680e-01,  ...,  2.7957e-01,\n",
       "            9.6500e-01, -7.0420e-01],\n",
       "          [ 4.7521e-01, -1.0617e-01, -5.2297e-01,  ..., -3.7392e-02,\n",
       "           -1.4795e-01, -5.7319e-01],\n",
       "          [ 4.8806e-01, -7.3731e-01, -2.7087e-01,  ..., -1.3114e-01,\n",
       "           -5.8920e-02, -2.7225e-01],\n",
       "          ...,\n",
       "          [ 3.7408e-01, -5.5616e-01, -1.1283e-02,  ..., -3.8965e-02,\n",
       "           -4.6020e-01, -3.0352e-01],\n",
       "          [ 1.5496e-01, -6.2912e-01, -5.6482e-01,  ..., -2.1157e-02,\n",
       "           -1.3097e-01, -3.8123e-01],\n",
       "          [ 6.2339e-01, -8.2271e-01, -2.5007e-01,  ..., -3.0012e-01,\n",
       "           -1.1233e-01,  2.3015e-02]],\n",
       "\n",
       "         [[ 5.0199e-01, -2.3120e-01, -7.1699e-02,  ..., -7.8622e-01,\n",
       "            8.2096e-01, -5.5458e-01],\n",
       "          [ 9.4219e-02,  2.2933e-01, -4.5380e-01,  ...,  5.3914e-01,\n",
       "           -5.4378e-01, -1.6946e-01],\n",
       "          [-7.8296e-02,  4.5119e-01, -2.2091e-01,  ...,  2.9916e-01,\n",
       "           -2.3367e-01, -5.3480e-01],\n",
       "          ...,\n",
       "          [-9.7827e-02, -1.6856e-01, -1.3667e-01,  ..., -2.1898e-02,\n",
       "           -1.5633e-01, -2.6846e-01],\n",
       "          [-3.0606e-02,  9.2354e-02, -2.0808e-01,  ..., -1.2131e-01,\n",
       "            1.2328e-01,  1.6442e-01],\n",
       "          [ 3.1249e-01,  4.4393e-01, -5.4824e-01,  ..., -6.0523e-01,\n",
       "           -3.0558e-01, -6.2880e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.7125e-01, -5.7419e-01, -4.5795e-02,  ..., -6.9443e-02,\n",
       "            4.5200e-01,  7.6607e-01],\n",
       "          [ 3.7919e-02, -3.8460e-01, -1.6165e-01,  ...,  5.1393e-01,\n",
       "            4.8335e-01,  6.2949e-01],\n",
       "          [ 2.8300e-01,  4.8353e-01,  6.2232e-02,  ...,  6.6391e-02,\n",
       "            2.4886e-01,  2.3912e-01],\n",
       "          ...,\n",
       "          [ 1.8039e-01, -2.0309e-01,  1.6088e-01,  ...,  5.2280e-01,\n",
       "            5.3835e-01,  2.8532e-01],\n",
       "          [ 1.7664e-01, -3.4490e-01,  3.9171e-01,  ...,  3.3453e-01,\n",
       "            6.7003e-01,  3.8024e-02],\n",
       "          [-3.3113e-02, -7.6127e-01,  2.4214e-01,  ...,  2.4490e-01,\n",
       "            5.4941e-01,  5.0322e-01]],\n",
       "\n",
       "         [[ 3.4879e-01,  9.7139e-01,  7.8166e-01,  ..., -2.1992e-02,\n",
       "            1.2881e-01,  5.4529e-01],\n",
       "          [ 3.5363e-01,  6.1914e-01, -1.8414e-01,  ..., -2.2530e-01,\n",
       "            1.2679e-01, -6.0664e-02],\n",
       "          [ 6.3583e-02,  7.9912e-01, -9.9400e-02,  ..., -1.8434e-01,\n",
       "            4.2215e-01,  8.0532e-02],\n",
       "          ...,\n",
       "          [ 2.7846e-01,  4.9016e-01, -4.8046e-01,  ..., -1.5691e-01,\n",
       "           -3.4376e-01, -3.8945e-01],\n",
       "          [ 8.6201e-01,  8.4453e-01, -2.0425e-01,  ...,  2.0963e-01,\n",
       "           -1.6564e-01,  2.4793e-02],\n",
       "          [ 7.4818e-01,  7.1503e-01, -5.8642e-01,  ...,  5.7996e-01,\n",
       "           -5.4168e-01, -2.0067e-02]],\n",
       "\n",
       "         [[-7.0975e-01, -2.2655e-01,  3.3875e-01,  ..., -3.6738e-02,\n",
       "           -1.2537e-01,  5.4769e-01],\n",
       "          [-4.5495e-01, -8.3780e-02, -4.7406e-01,  ..., -2.6290e-01,\n",
       "           -6.7799e-02,  3.2502e-01],\n",
       "          [-6.9618e-01,  7.4213e-03, -2.9114e-01,  ..., -1.2843e-01,\n",
       "            1.8130e-01,  1.0684e-01],\n",
       "          ...,\n",
       "          [-5.7701e-01, -3.8397e-02, -1.2049e-01,  ..., -2.5702e-01,\n",
       "            1.2675e-02,  9.8197e-02],\n",
       "          [-5.4655e-01,  2.7782e-04,  4.5738e-02,  ..., -1.0425e-01,\n",
       "           -1.9503e-01,  3.7199e-01],\n",
       "          [-6.7102e-01,  2.6388e-01,  1.0925e+00,  ..., -5.1660e-01,\n",
       "           -6.9863e-01,  4.7437e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.4533e+00,  4.7046e-01, -9.2651e-01,  ...,  4.0429e-01,\n",
       "           -4.9602e-01, -8.1135e-01],\n",
       "          [-1.0576e+00,  1.9996e-01,  7.9382e-02,  ...,  3.1507e-01,\n",
       "           -1.2037e+00, -1.0639e+00],\n",
       "          [-3.5095e-01,  2.0398e-01, -1.3905e-01,  ...,  7.1160e-02,\n",
       "           -5.3144e-01, -2.9457e-01],\n",
       "          ...,\n",
       "          [-3.7693e-01,  3.3828e-01, -2.6030e-01,  ...,  3.6927e-01,\n",
       "           -6.3680e-01, -1.5446e-01],\n",
       "          [-5.4908e-01,  3.7414e-02, -2.6793e-01,  ...,  4.0771e-01,\n",
       "           -7.5033e-01, -1.0873e-01],\n",
       "          [-2.8557e-01,  5.0279e-01, -3.4852e-01,  ..., -1.9447e-01,\n",
       "           -4.4557e-01, -2.4202e-01]],\n",
       "\n",
       "         [[ 9.6895e-01,  1.8554e-01, -3.8688e-01,  ...,  2.7211e-01,\n",
       "            1.0017e+00, -6.8701e-01],\n",
       "          [ 4.2615e-01, -4.6821e-01, -8.3329e-01,  ...,  5.3144e-01,\n",
       "            7.0055e-02, -5.9724e-01],\n",
       "          [ 4.5670e-01, -2.2997e-01, -1.6970e-01,  ...,  2.6323e-03,\n",
       "            1.7371e-01, -4.0180e-02],\n",
       "          ...,\n",
       "          [ 1.9721e-01, -2.7222e-01, -2.7507e-01,  ...,  1.1273e-01,\n",
       "           -7.4241e-02, -4.3863e-01],\n",
       "          [ 5.3175e-01, -4.8080e-03,  3.6882e-02,  ..., -5.2962e-02,\n",
       "            2.4841e-01, -2.4030e-01],\n",
       "          [ 7.1277e-01, -3.5056e-01, -1.7297e-01,  ..., -1.2060e-01,\n",
       "            4.0546e-01,  2.0991e-01]],\n",
       "\n",
       "         [[ 5.0168e-01, -2.0971e-01, -7.5823e-02,  ..., -7.7499e-01,\n",
       "            8.4356e-01, -5.5903e-01],\n",
       "          [ 8.9633e-02, -3.7190e-01, -3.6400e-01,  ...,  1.3064e-01,\n",
       "           -4.5859e-01, -2.0698e-01],\n",
       "          [-1.8624e-01, -2.2060e-01,  2.6376e-01,  ..., -2.7682e-01,\n",
       "            2.8313e-01, -4.7898e-01],\n",
       "          ...,\n",
       "          [-9.0819e-02,  2.8642e-01,  4.2323e-02,  ...,  4.5679e-01,\n",
       "            1.7375e-01, -3.2435e-01],\n",
       "          [ 3.9279e-02, -1.1975e-01, -2.7674e-01,  ..., -2.8930e-01,\n",
       "            2.6363e-01, -1.0408e-01],\n",
       "          [ 4.2419e-01,  2.8343e-01, -1.8484e-01,  ..., -1.4224e-01,\n",
       "           -1.4092e-02, -1.2537e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.0356e-01, -5.8341e-01, -2.3898e-02,  ..., -6.9242e-02,\n",
       "            4.6070e-01,  8.1518e-01],\n",
       "          [-1.2655e-01, -1.2442e-01,  2.7679e-01,  ...,  6.6205e-01,\n",
       "           -1.2068e-01,  3.9546e-01],\n",
       "          [ 1.3029e-01, -5.1504e-02,  4.3009e-01,  ...,  4.1666e-01,\n",
       "            4.4339e-01,  1.1083e-01],\n",
       "          ...,\n",
       "          [-2.1141e-01, -5.8939e-01,  4.0289e-01,  ...,  2.2912e-01,\n",
       "            1.7518e-01,  3.4550e-01],\n",
       "          [ 1.8468e-01, -1.0676e-01,  3.1271e-01,  ...,  2.4158e-01,\n",
       "            6.7589e-01,  4.5098e-01],\n",
       "          [ 8.5838e-02, -2.9172e-01, -6.2418e-02,  ..., -1.2744e-01,\n",
       "            5.8267e-01,  5.7502e-01]],\n",
       "\n",
       "         [[ 3.3265e-01,  9.2326e-01,  8.4779e-01,  ..., -7.0526e-02,\n",
       "            1.2519e-01,  5.6326e-01],\n",
       "          [ 5.4056e-01,  4.6169e-01, -4.4035e-01,  ...,  1.5516e-01,\n",
       "           -3.4196e-01,  1.3042e-01],\n",
       "          [ 9.6889e-02,  5.6922e-01,  1.3109e-01,  ...,  9.5364e-02,\n",
       "           -6.5983e-03,  1.6875e-01],\n",
       "          ...,\n",
       "          [ 3.2063e-01,  1.7792e-01, -4.8664e-01,  ..., -9.7230e-02,\n",
       "           -3.1037e-01,  1.9396e-01],\n",
       "          [ 4.0274e-01,  2.8175e-01,  2.9517e-03,  ..., -3.5513e-01,\n",
       "           -4.1544e-03, -5.8997e-02],\n",
       "          [ 7.2097e-01,  2.6444e-01, -3.5350e-01,  ...,  4.6122e-01,\n",
       "           -2.9124e-01, -2.9239e-01]],\n",
       "\n",
       "         [[-7.4424e-01, -2.0886e-01,  3.8702e-01,  ...,  1.7878e-02,\n",
       "           -1.1575e-01,  5.9777e-01],\n",
       "          [-6.9563e-01, -5.8093e-01,  4.8639e-02,  ..., -2.7642e-01,\n",
       "           -6.5248e-01,  7.5776e-01],\n",
       "          [-4.8092e-01,  1.5282e-01,  1.3907e-02,  ..., -1.5945e-01,\n",
       "           -1.5480e-01,  8.2484e-01],\n",
       "          ...,\n",
       "          [-1.2860e+00, -4.1399e-01,  1.9689e-01,  ..., -3.0978e-01,\n",
       "           -6.1396e-01,  4.0089e-01],\n",
       "          [-1.0657e+00, -3.4473e-02, -5.3754e-01,  ...,  2.1465e-01,\n",
       "            4.6710e-01, -1.7130e-01],\n",
       "          [-5.4963e-01, -2.7909e-01, -9.0543e-02,  ..., -1.1628e-01,\n",
       "            7.2133e-02,  2.7361e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.4357e+00,  4.7249e-01, -9.2323e-01,  ...,  3.8371e-01,\n",
       "           -4.7958e-01, -8.2065e-01],\n",
       "          [-7.5243e-01,  4.6241e-01, -6.9417e-01,  ...,  5.9033e-01,\n",
       "           -3.2371e-01, -8.3758e-01],\n",
       "          [-5.6100e-01,  3.7620e-01, -6.4700e-01,  ...,  2.4649e-01,\n",
       "           -6.7631e-01, -6.2200e-01],\n",
       "          ...,\n",
       "          [-1.0605e-01,  2.0265e-01, -2.8887e-01,  ...,  3.4322e-02,\n",
       "           -3.5267e-01, -2.6328e-01],\n",
       "          [-5.4257e-01,  2.7781e-01, -2.1747e-01,  ..., -4.9956e-02,\n",
       "           -5.9313e-01,  6.3357e-02],\n",
       "          [ 3.7708e-01,  2.5399e-01, -6.6274e-01,  ..., -1.1164e-01,\n",
       "           -5.0485e-01, -1.6406e-01]],\n",
       "\n",
       "         [[ 9.7028e-01,  1.5608e-01, -4.0191e-01,  ...,  2.8473e-01,\n",
       "            1.0077e+00, -6.8284e-01],\n",
       "          [ 5.5343e-01, -4.2520e-01, -4.0097e-01,  ...,  1.5067e-01,\n",
       "            6.1570e-01, -3.5387e-01],\n",
       "          [ 4.2077e-01, -6.4045e-01, -2.7108e-01,  ...,  3.4112e-01,\n",
       "            6.8311e-01, -3.9479e-01],\n",
       "          ...,\n",
       "          [ 5.5539e-01, -7.4307e-01, -4.6424e-01,  ...,  3.7660e-01,\n",
       "            1.8518e-01, -6.5989e-01],\n",
       "          [ 1.9957e-01, -4.8904e-01, -6.1837e-01,  ...,  2.2389e-02,\n",
       "            1.9489e-01,  2.5081e-02],\n",
       "          [ 5.5966e-01, -5.0847e-01, -3.8651e-02,  ...,  6.6388e-02,\n",
       "            5.5057e-01, -2.6682e-01]],\n",
       "\n",
       "         [[ 4.7820e-01, -2.2156e-01, -8.9745e-02,  ..., -7.6083e-01,\n",
       "            8.3794e-01, -5.5256e-01],\n",
       "          [ 2.4996e-01, -3.0420e-01, -4.7652e-01,  ...,  2.3649e-01,\n",
       "            2.4902e-01, -1.8115e-01],\n",
       "          [ 4.1677e-01, -1.2706e-01, -1.6503e-01,  ...,  3.1021e-01,\n",
       "            3.7837e-01, -2.6469e-02],\n",
       "          ...,\n",
       "          [ 4.9196e-01, -1.1502e-01, -3.4645e-01,  ..., -1.1767e-02,\n",
       "            1.7687e-02, -1.8225e-01],\n",
       "          [ 3.1369e-01,  7.8946e-02, -5.4752e-01,  ...,  5.1278e-02,\n",
       "            1.0071e-01, -3.3157e-01],\n",
       "          [ 5.3112e-01,  4.8174e-01, -1.5178e-01,  ..., -4.4580e-01,\n",
       "           -2.3468e-01,  1.8763e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.8591e-01, -5.9690e-01, -3.1390e-02,  ..., -7.7890e-02,\n",
       "            4.7084e-01,  7.8985e-01],\n",
       "          [ 2.9939e-01, -3.7695e-01,  3.5466e-01,  ..., -1.5907e-01,\n",
       "            5.8154e-01,  1.5131e-01],\n",
       "          [-2.4696e-01, -5.9025e-02,  1.0662e-01,  ...,  1.7535e-01,\n",
       "            8.7100e-01,  8.8440e-01],\n",
       "          ...,\n",
       "          [ 2.2360e-01, -5.5894e-01, -4.0172e-01,  ...,  2.3004e-01,\n",
       "            5.5774e-01,  1.7598e-01],\n",
       "          [ 1.9716e-01, -4.0661e-01,  3.2079e-01,  ...,  1.4918e-01,\n",
       "            4.3968e-01,  7.5465e-01],\n",
       "          [ 3.6223e-01, -8.0702e-01,  4.2428e-01,  ...,  1.2853e-01,\n",
       "            3.1454e-01,  3.1087e-01]],\n",
       "\n",
       "         [[ 3.4234e-01,  9.3661e-01,  8.3152e-01,  ..., -5.3702e-02,\n",
       "            1.3273e-01,  5.7934e-01],\n",
       "          [ 4.8595e-01,  7.3988e-01,  4.8365e-01,  ...,  4.8291e-01,\n",
       "            1.9078e-01,  8.6409e-02],\n",
       "          [ 9.1992e-01,  2.7398e-01,  2.8752e-01,  ...,  3.4770e-01,\n",
       "            1.7640e-01, -7.2404e-02],\n",
       "          ...,\n",
       "          [ 3.8903e-01,  8.5761e-01,  1.3751e-01,  ...,  4.5615e-02,\n",
       "           -2.4694e-01,  1.0233e-01],\n",
       "          [ 7.5547e-01,  3.6646e-01, -1.9030e-01,  ...,  3.4542e-01,\n",
       "           -1.0693e-01, -1.8930e-01],\n",
       "          [ 7.4205e-01,  5.6533e-01, -7.2099e-01,  ...,  9.5869e-01,\n",
       "           -3.7051e-01, -3.7626e-01]],\n",
       "\n",
       "         [[-7.3332e-01, -2.3560e-01,  3.6046e-01,  ...,  7.3529e-03,\n",
       "           -1.1866e-01,  5.6810e-01],\n",
       "          [-3.9297e-01, -6.7017e-02, -1.6232e-02,  ..., -3.2880e-01,\n",
       "           -1.8281e-01,  4.2607e-01],\n",
       "          [-7.3531e-01, -7.7737e-02,  1.0193e-01,  ...,  3.0303e-01,\n",
       "           -1.7762e-01,  8.8112e-02],\n",
       "          ...,\n",
       "          [-5.7605e-01, -4.1934e-02, -1.1574e-01,  ..., -1.0441e-01,\n",
       "           -4.6408e-01, -1.2991e-01],\n",
       "          [-7.9728e-01, -1.2412e-01, -2.2193e-02,  ...,  2.2415e-01,\n",
       "           -1.8803e-01, -5.9296e-02],\n",
       "          [-5.6784e-01,  2.1946e-02,  3.6733e-01,  ...,  9.3855e-02,\n",
       "           -2.8269e-01,  3.9635e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-2.4241e-01, -3.9387e-01,  4.8092e-02,  ...,  1.1447e-01,\n",
       "           -1.4142e-02,  7.2601e-01],\n",
       "          [ 1.2910e-01,  2.2974e-01,  7.0239e-02,  ..., -6.5430e-02,\n",
       "            1.7231e-02,  5.6933e-01],\n",
       "          [-5.1720e-01, -9.9938e-03,  1.1272e-01,  ..., -8.5706e-02,\n",
       "           -4.3588e-02,  8.0257e-01],\n",
       "          ...,\n",
       "          [ 4.2465e-01, -4.2145e-01, -5.0099e-02,  ..., -1.1066e-01,\n",
       "            2.1973e-01,  2.6670e-01],\n",
       "          [ 2.2418e-01,  2.2323e-01, -2.3365e-02,  ..., -1.0311e-04,\n",
       "           -1.9103e-01,  7.5365e-01],\n",
       "          [ 3.1106e-02, -8.9935e-02, -8.5808e-01,  ..., -7.0639e-02,\n",
       "           -6.2193e-01,  1.2109e+00]],\n",
       "\n",
       "         [[ 7.6021e-02,  2.2589e-01,  2.3859e-02,  ...,  1.7156e-01,\n",
       "           -2.0336e-02, -2.5439e-01],\n",
       "          [-1.1785e-01,  3.0234e-01, -3.1444e-02,  ...,  4.7852e-01,\n",
       "            2.4860e-01, -3.1819e-01],\n",
       "          [-1.3527e-01,  1.8902e-01, -9.4949e-02,  ...,  3.8646e-01,\n",
       "            6.2162e-01, -2.7416e-01],\n",
       "          ...,\n",
       "          [-2.6346e-01,  1.7752e-01, -1.4552e-03,  ..., -2.9956e-01,\n",
       "            2.7034e-01, -2.0634e-01],\n",
       "          [-3.7032e-01,  6.2965e-01, -2.8617e-01,  ..., -1.0242e-01,\n",
       "            7.6422e-02,  1.8771e-01],\n",
       "          [-2.9760e-01,  6.6472e-01, -4.6947e-01,  ..., -9.8787e-01,\n",
       "            8.3851e-01,  3.7814e-01]],\n",
       "\n",
       "         [[ 7.3778e-01, -1.5195e-01,  6.5394e-01,  ...,  5.6289e-01,\n",
       "           -3.6926e-02, -5.4639e-01],\n",
       "          [ 2.6366e-01, -4.0655e-01,  4.7955e-01,  ...,  1.7269e-01,\n",
       "            9.3346e-02, -7.9097e-01],\n",
       "          [ 4.8204e-01, -6.1996e-01,  4.6201e-01,  ...,  4.3541e-01,\n",
       "           -5.5952e-02, -6.5321e-01],\n",
       "          ...,\n",
       "          [ 7.9076e-02, -2.5533e-01,  6.4608e-01,  ..., -2.0667e-01,\n",
       "            6.5603e-02, -9.6040e-01],\n",
       "          [ 3.5781e-01, -5.0750e-01,  4.0517e-01,  ...,  3.7865e-01,\n",
       "           -2.0774e-02, -7.9297e-01],\n",
       "          [ 8.3846e-01, -6.1606e-01,  3.7881e-02,  ...,  1.2642e+00,\n",
       "           -3.2923e-01, -8.5519e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.7059e-01, -4.2006e-01, -1.1451e-01,  ..., -2.0903e-01,\n",
       "            9.1282e-02,  9.8591e-02],\n",
       "          [-5.3626e-03, -2.3216e-01,  2.3177e-01,  ..., -1.5560e-01,\n",
       "           -4.2813e-01, -1.4742e-01],\n",
       "          [ 5.7933e-02, -6.4427e-02,  2.1994e-01,  ...,  8.0455e-02,\n",
       "           -1.9493e-01,  1.5629e-01],\n",
       "          ...,\n",
       "          [-2.8419e-01,  1.5983e-01, -4.2583e-01,  ...,  4.6204e-01,\n",
       "           -3.5417e-01,  2.5483e-01],\n",
       "          [ 2.3239e-01, -1.7776e-01, -2.9083e-01,  ...,  5.0995e-01,\n",
       "           -8.9658e-02,  5.6114e-01],\n",
       "          [-8.8728e-02, -8.7424e-01, -3.7074e-01,  ...,  1.0527e+00,\n",
       "           -2.2848e-02,  9.3699e-01]],\n",
       "\n",
       "         [[-6.0606e-01, -2.7061e-01,  2.2306e-01,  ..., -5.6532e-02,\n",
       "           -1.6673e-01, -2.0687e-01],\n",
       "          [ 4.5158e-02, -9.9288e-02, -6.5264e-01,  ..., -1.1921e-01,\n",
       "           -3.9452e-01, -1.2455e-01],\n",
       "          [-3.0741e-01, -1.0249e-01, -1.4565e-01,  ..., -1.5201e-01,\n",
       "           -3.4353e-01,  7.5070e-02],\n",
       "          ...,\n",
       "          [-2.1566e-01, -1.3851e-01, -2.2921e-02,  ..., -2.5426e-01,\n",
       "           -4.9725e-02, -4.0026e-02],\n",
       "          [-6.8727e-01, -4.2681e-01,  2.2206e-01,  ..., -1.5851e-01,\n",
       "           -2.3255e-01,  2.6356e-02],\n",
       "          [-1.0575e+00,  2.4611e-01, -3.2782e-01,  ..., -5.1571e-02,\n",
       "           -1.3749e-01, -4.6381e-02]],\n",
       "\n",
       "         [[ 2.5894e-01,  4.4587e-02, -5.5796e-01,  ...,  7.8662e-02,\n",
       "           -3.2258e-02, -1.4064e-01],\n",
       "          [ 2.7520e-02, -1.1305e-01, -2.1431e-01,  ...,  5.5917e-02,\n",
       "            2.2707e-01,  1.5626e-01],\n",
       "          [-3.1062e-01, -3.2701e-01, -3.5223e-02,  ..., -1.7536e-01,\n",
       "            1.6008e-01, -1.9350e-01],\n",
       "          ...,\n",
       "          [-1.3486e-01,  1.1712e-01, -5.0768e-02,  ..., -8.8128e-02,\n",
       "            4.4972e-01, -3.2826e-01],\n",
       "          [ 1.1507e-02, -9.2349e-02, -3.7581e-01,  ...,  2.3972e-01,\n",
       "            6.4001e-01, -3.9803e-01],\n",
       "          [ 3.9407e-01, -4.3888e-01, -5.3460e-01,  ...,  1.9466e-01,\n",
       "            8.4034e-01, -4.3519e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.2739e-01, -4.1478e-01,  1.4071e-02,  ...,  1.0272e-01,\n",
       "            1.3198e-02,  7.2148e-01],\n",
       "          [-3.1356e-01,  4.8963e-03,  8.1288e-02,  ..., -6.8392e-01,\n",
       "           -9.3858e-02,  6.9401e-01],\n",
       "          [-5.0228e-01, -2.8882e-01,  2.4471e-01,  ..., -2.1156e-01,\n",
       "            8.8825e-03,  6.6038e-01],\n",
       "          ...,\n",
       "          [-1.5225e-01, -2.1913e-01, -6.3323e-01,  ..., -2.5077e-01,\n",
       "            5.1250e-01,  2.7760e-01],\n",
       "          [-3.2829e-02,  2.9858e-02,  6.7040e-02,  ...,  4.7413e-02,\n",
       "            3.0848e-01,  6.1959e-01],\n",
       "          [-4.0883e-01, -3.8723e-01, -6.4569e-01,  ...,  8.2381e-03,\n",
       "           -1.1639e-01,  6.8074e-01]],\n",
       "\n",
       "         [[ 1.1426e-01,  2.3661e-01,  2.6923e-02,  ...,  1.3664e-01,\n",
       "           -4.7514e-02, -2.4557e-01],\n",
       "          [-3.1298e-03,  6.8830e-01,  2.3236e-01,  ...,  4.2227e-01,\n",
       "            1.5120e-01, -2.6478e-01],\n",
       "          [ 2.3965e-01,  6.8646e-01,  1.6865e-01,  ...,  2.6165e-02,\n",
       "           -1.4574e-01, -1.0187e-01],\n",
       "          ...,\n",
       "          [ 7.6686e-02,  1.4412e-01, -5.8765e-03,  ..., -9.1116e-03,\n",
       "           -3.4063e-01, -2.8793e-01],\n",
       "          [ 3.4081e-01,  3.2936e-01, -6.2013e-02,  ..., -2.0172e-01,\n",
       "           -5.9654e-03,  7.1015e-02],\n",
       "          [ 3.9228e-02,  6.2064e-01, -4.5073e-01,  ..., -5.9856e-01,\n",
       "            8.0788e-02,  6.4231e-02]],\n",
       "\n",
       "         [[ 6.9868e-01, -1.5890e-01,  7.0473e-01,  ...,  5.7105e-01,\n",
       "           -3.7524e-03, -5.9011e-01],\n",
       "          [ 5.2278e-01, -4.8656e-01,  5.2675e-01,  ...,  3.6852e-01,\n",
       "            1.2701e-01, -1.0667e+00],\n",
       "          [-1.0058e-02, -7.2100e-01,  4.7413e-01,  ...,  4.6975e-01,\n",
       "           -1.3385e-01, -5.3788e-01],\n",
       "          ...,\n",
       "          [ 1.0770e-02, -8.1737e-01,  8.5244e-01,  ...,  9.6991e-02,\n",
       "           -1.8225e-02, -1.4070e+00],\n",
       "          [ 1.1158e-01, -1.5135e-01,  4.8259e-01,  ...,  1.2542e-02,\n",
       "            1.5608e-01, -8.4859e-01],\n",
       "          [ 5.0914e-01, -2.8506e-01,  3.0882e-01,  ...,  6.3462e-01,\n",
       "           -7.1513e-03, -6.6299e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.2292e-01, -4.5862e-01, -1.5148e-01,  ..., -2.4539e-01,\n",
       "            1.2781e-01,  1.0677e-01],\n",
       "          [ 3.3226e-01, -2.9235e-01, -4.3820e-01,  ..., -1.5304e-01,\n",
       "           -1.1218e+00,  3.5672e-01],\n",
       "          [-1.8197e-01, -4.6749e-01, -2.9158e-01,  ..., -1.1026e-01,\n",
       "           -6.5419e-01,  4.8677e-01],\n",
       "          ...,\n",
       "          [ 5.2710e-01,  1.6166e-01, -7.4081e-01,  ..., -2.4262e-01,\n",
       "           -4.6703e-01,  4.0057e-01],\n",
       "          [-5.7852e-02, -6.3623e-01, -2.4399e-01,  ..., -4.0322e-01,\n",
       "           -1.8928e-01,  3.0043e-01],\n",
       "          [-1.2298e-01, -6.0277e-01, -8.2768e-01,  ...,  3.0389e-01,\n",
       "            1.4116e-01,  4.6541e-01]],\n",
       "\n",
       "         [[-5.9970e-01, -2.6376e-01,  1.9401e-01,  ..., -5.3214e-02,\n",
       "           -1.4177e-01, -1.8514e-01],\n",
       "          [-2.6488e-01,  2.6850e-01, -4.8643e-02,  ..., -3.8296e-01,\n",
       "           -3.3388e-01, -7.7450e-02],\n",
       "          [-4.7410e-01,  8.9708e-02,  2.9297e-01,  ..., -8.3871e-02,\n",
       "           -4.2229e-01, -2.5640e-02],\n",
       "          ...,\n",
       "          [-2.5134e-01, -2.2411e-01, -1.8789e-01,  ..., -3.6321e-01,\n",
       "           -1.1923e-01,  5.1641e-01],\n",
       "          [-4.1591e-01, -2.7895e-01, -1.6456e-01,  ...,  4.3538e-02,\n",
       "            7.0749e-02,  2.2935e-01],\n",
       "          [-3.0869e-01, -1.3707e-01, -3.5974e-01,  ...,  3.3742e-02,\n",
       "           -1.8373e-01, -1.4468e-01]],\n",
       "\n",
       "         [[ 2.4842e-01,  3.5036e-02, -5.9320e-01,  ...,  9.2580e-02,\n",
       "           -4.0867e-02, -1.4249e-01],\n",
       "          [ 2.7486e-01, -4.0555e-01, -4.2111e-01,  ..., -2.2552e-01,\n",
       "            6.3996e-01, -2.0547e-01],\n",
       "          [ 2.6118e-01, -1.0687e-01, -1.7680e-01,  ...,  3.5207e-01,\n",
       "            6.2515e-01, -1.6508e-01],\n",
       "          ...,\n",
       "          [ 1.4737e-02,  1.0664e-01, -6.4995e-01,  ...,  1.7366e-01,\n",
       "            1.4104e-01, -1.9846e-01],\n",
       "          [-3.1159e-01, -1.6544e-01, -7.8741e-02,  ...,  4.6519e-01,\n",
       "            4.8206e-01,  6.6425e-02],\n",
       "          [-1.8203e-01, -5.5515e-01, -1.0929e-01,  ...,  1.9478e-01,\n",
       "            6.3258e-01, -4.2075e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.2844e-01, -4.1739e-01,  1.2770e-02,  ...,  1.0685e-01,\n",
       "           -7.2169e-03,  7.4196e-01],\n",
       "          [-1.1595e-01,  5.2341e-02, -1.1588e-01,  ...,  1.7676e-01,\n",
       "           -1.5010e-01,  7.8639e-01],\n",
       "          [ 1.6182e-01, -3.3924e-02, -1.8070e-01,  ...,  8.0147e-02,\n",
       "           -2.2276e-01,  7.1644e-01],\n",
       "          ...,\n",
       "          [ 3.4729e-01,  1.5800e-01, -2.7763e-01,  ...,  6.1524e-03,\n",
       "            4.2948e-03,  4.9789e-01],\n",
       "          [-2.4059e-01,  6.4744e-02, -2.0146e-01,  ..., -1.3063e-01,\n",
       "           -1.6432e-01,  6.5533e-01],\n",
       "          [ 1.8259e-01, -3.7416e-01, -7.7377e-01,  ...,  1.8830e-01,\n",
       "           -4.5262e-01,  6.6046e-01]],\n",
       "\n",
       "         [[ 9.9456e-02,  2.3789e-01,  9.6863e-03,  ...,  1.4439e-01,\n",
       "           -2.8633e-02, -2.4772e-01],\n",
       "          [-2.6843e-01,  4.7067e-01, -1.9873e-01,  ..., -1.2093e-01,\n",
       "            2.1540e-01, -1.3801e-01],\n",
       "          [-1.1381e-01,  6.0266e-01, -3.7390e-01,  ..., -2.0491e-01,\n",
       "            7.9210e-01, -1.0516e-01],\n",
       "          ...,\n",
       "          [-1.8220e-01,  6.6375e-01,  1.1608e-01,  ..., -3.8310e-01,\n",
       "            3.4255e-01, -1.1411e-01],\n",
       "          [ 6.9981e-02,  4.2504e-01, -1.2559e-01,  ..., -4.1757e-01,\n",
       "            2.3700e-01, -7.3466e-02],\n",
       "          [-7.6950e-01,  3.7638e-01, -3.8415e-01,  ..., -7.9293e-01,\n",
       "            6.5650e-01,  3.7323e-01]],\n",
       "\n",
       "         [[ 7.1273e-01, -1.4059e-01,  7.0203e-01,  ...,  5.7938e-01,\n",
       "           -5.4103e-03, -5.7058e-01],\n",
       "          [-1.4057e-01, -1.1445e-01,  7.1048e-01,  ...,  8.7755e-01,\n",
       "            1.3934e-01, -9.0509e-01],\n",
       "          [-1.2925e-01, -2.1016e-01,  7.8258e-01,  ...,  6.7129e-01,\n",
       "            2.3452e-01, -6.7539e-01],\n",
       "          ...,\n",
       "          [ 1.8616e-01, -1.2318e-01,  5.1765e-01,  ...,  2.2647e-01,\n",
       "           -1.1598e-01, -8.0248e-01],\n",
       "          [ 2.4201e-01, -2.1520e-01,  4.8095e-01,  ...,  3.3173e-01,\n",
       "            6.1419e-02, -5.0064e-01],\n",
       "          [ 1.0188e+00, -3.0067e-02,  1.9166e-02,  ...,  7.3587e-01,\n",
       "           -4.8246e-01, -1.3351e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.7573e-01, -4.8228e-01, -1.3167e-01,  ..., -2.2454e-01,\n",
       "            1.0866e-01,  8.4360e-02],\n",
       "          [-1.7957e-01, -3.9441e-01,  6.2021e-02,  ...,  2.8498e-01,\n",
       "            1.5177e-01, -6.7532e-03],\n",
       "          [-3.0798e-02, -6.3550e-01, -1.1695e-01,  ...,  5.2471e-02,\n",
       "            7.2746e-02,  6.9117e-02],\n",
       "          ...,\n",
       "          [-2.2710e-01, -6.4640e-01,  1.0066e-01,  ...,  4.3612e-01,\n",
       "           -2.7429e-01, -1.6268e-01],\n",
       "          [ 2.0771e-01, -3.5866e-01, -5.3660e-01,  ...,  2.5716e-01,\n",
       "           -4.9874e-02,  4.1789e-01],\n",
       "          [-2.0942e-01, -7.3405e-01, -7.9610e-01,  ...,  9.8379e-01,\n",
       "            4.0636e-01,  6.2454e-01]],\n",
       "\n",
       "         [[-6.0889e-01, -2.7499e-01,  1.7052e-01,  ..., -5.1757e-02,\n",
       "           -1.2535e-01, -1.8241e-01],\n",
       "          [-7.0848e-01, -1.9415e-01, -2.1813e-01,  ..., -3.5801e-01,\n",
       "           -1.9995e-01,  3.1691e-01],\n",
       "          [-7.6162e-01, -1.9114e-01, -3.2238e-01,  ..., -9.4654e-02,\n",
       "           -1.3547e-01,  3.5544e-01],\n",
       "          ...,\n",
       "          [-3.1874e-01, -9.7321e-02,  5.8356e-02,  ..., -4.6582e-01,\n",
       "           -1.7252e-01,  5.5489e-02],\n",
       "          [-3.9087e-01, -5.9421e-01, -1.9274e-01,  ..., -1.4990e-01,\n",
       "           -2.3022e-01, -1.6093e-02],\n",
       "          [-4.2960e-01,  1.9698e-01, -4.9341e-02,  ...,  1.5221e-01,\n",
       "           -8.3071e-02, -6.8276e-02]],\n",
       "\n",
       "         [[ 2.5084e-01,  4.9145e-02, -5.8485e-01,  ...,  7.3736e-02,\n",
       "           -3.9885e-02, -1.3424e-01],\n",
       "          [-4.1531e-02,  3.6694e-02, -5.2964e-01,  ..., -6.0106e-03,\n",
       "            7.7752e-01, -3.3750e-01],\n",
       "          [ 4.0135e-01,  9.3680e-03, -5.9775e-01,  ...,  3.2273e-01,\n",
       "            7.3568e-01, -3.2937e-01],\n",
       "          ...,\n",
       "          [ 6.9179e-02,  9.5698e-02, -4.9161e-01,  ..., -8.3985e-02,\n",
       "            1.0301e+00, -2.2888e-01],\n",
       "          [ 4.6520e-01, -1.8391e-01, -4.3062e-01,  ...,  2.6640e-01,\n",
       "            8.9966e-01, -2.8498e-01],\n",
       "          [-1.9904e-01, -5.2827e-01, -2.6820e-01,  ..., -8.4047e-02,\n",
       "            8.3728e-01, -4.0152e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 2.0378e+00, -1.2829e+00,  1.7811e+00,  ...,  8.7093e-01,\n",
       "            2.4961e+00, -3.4435e+00],\n",
       "          [-1.5371e+00, -7.3594e-01,  2.2323e+00,  ..., -5.9739e-01,\n",
       "            1.5081e+00,  8.2439e-01],\n",
       "          [-2.0314e+00, -1.4665e+00, -7.4647e-02,  ..., -3.4328e-01,\n",
       "            1.2469e+00, -4.5548e-01],\n",
       "          ...,\n",
       "          [ 5.3332e-01,  2.8428e+00,  2.4027e-01,  ..., -2.1543e+00,\n",
       "            1.2162e+00,  4.9930e+00],\n",
       "          [ 9.0175e-01,  5.4534e-03,  7.3636e-01,  ..., -1.0473e+00,\n",
       "           -2.8736e-02,  5.0981e-01],\n",
       "          [-2.6519e+00, -7.3122e-01,  3.9799e+00,  ...,  1.6109e+00,\n",
       "           -4.8702e-01, -1.4333e-01]],\n",
       "\n",
       "         [[ 1.2373e+00,  2.7576e+00,  1.1718e-01,  ...,  1.8775e-01,\n",
       "            3.2091e-01,  2.3144e+00],\n",
       "          [-2.6945e+00,  1.9920e+00,  2.1792e+00,  ...,  1.6339e+00,\n",
       "           -4.7595e-01, -2.9020e+00],\n",
       "          [-1.4503e+00, -4.5632e-01,  5.0886e+00,  ...,  3.4331e-01,\n",
       "            8.5622e-01,  4.4982e-01],\n",
       "          ...,\n",
       "          [-4.2577e+00,  2.2174e+00,  8.7886e-01,  ...,  1.6720e+00,\n",
       "           -8.4256e-01, -9.5267e-01],\n",
       "          [-2.4256e+00,  2.0836e+00,  1.1057e+00,  ...,  1.1720e+00,\n",
       "            1.9611e+00,  9.6643e-01],\n",
       "          [ 1.4464e+00, -9.3397e-01,  3.8858e+00,  ...,  7.6410e-01,\n",
       "           -1.0678e+00,  1.2952e+00]],\n",
       "\n",
       "         [[-1.0787e+00,  2.9533e-01, -3.2893e-01,  ...,  1.2807e+00,\n",
       "           -1.2500e+00, -1.1876e+00],\n",
       "          [-1.6093e+00, -1.0093e+00, -2.9416e+00,  ...,  1.2587e-01,\n",
       "            1.0183e+00, -2.2256e+00],\n",
       "          [-1.6535e+00, -9.6575e-01, -4.8630e+00,  ...,  7.3031e-01,\n",
       "            1.5431e+00, -7.5446e-01],\n",
       "          ...,\n",
       "          [-5.9741e-01,  1.8266e+00, -2.7271e-01,  ..., -1.5331e+00,\n",
       "            1.0856e+00, -1.6889e-01],\n",
       "          [ 1.7740e-01,  4.2520e-01, -4.1514e-01,  ..., -4.5637e-01,\n",
       "            8.5292e-01,  3.8656e-01],\n",
       "          [-8.3390e-01,  9.7705e-01, -3.6055e+00,  ...,  1.0854e+00,\n",
       "           -5.9692e-01,  7.7123e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-7.8159e-01, -2.3474e-01,  3.1208e-01,  ..., -1.6873e+00,\n",
       "           -2.9248e-02,  4.3623e-01],\n",
       "          [-1.1355e+00, -3.2920e-01,  1.3699e+00,  ..., -3.0285e+00,\n",
       "           -1.3272e+00,  1.7430e+00],\n",
       "          [-1.8123e+00,  1.0586e+00,  8.1230e-01,  ..., -3.4185e+00,\n",
       "           -1.6059e-01,  3.0651e+00],\n",
       "          ...,\n",
       "          [-1.5512e+00,  1.7433e+00, -6.1750e-01,  ..., -2.1635e+00,\n",
       "           -1.2278e+00, -9.8277e-01],\n",
       "          [-1.3770e+00, -9.9844e-01,  4.9647e-01,  ..., -7.0121e-01,\n",
       "           -7.7951e-02, -1.0840e+00],\n",
       "          [ 1.9179e-01, -2.5596e+00, -3.1835e+00,  ...,  3.5241e+00,\n",
       "            6.4460e-01, -2.0947e+00]],\n",
       "\n",
       "         [[ 7.5126e-01, -1.5234e+00, -1.7164e+00,  ...,  1.2027e-01,\n",
       "           -5.2581e-01, -5.5622e-01],\n",
       "          [ 1.2441e+00, -3.9375e-01, -2.2133e+00,  ..., -1.4334e+00,\n",
       "            3.1619e-01, -6.6323e-01],\n",
       "          [-1.3997e+00,  2.6017e+00, -2.2731e+00,  ..., -5.4221e-01,\n",
       "           -3.0057e-01, -7.6968e-02],\n",
       "          ...,\n",
       "          [-3.6056e+00, -2.8263e-01, -2.6524e+00,  ..., -9.9588e-01,\n",
       "            3.8466e-01, -2.5556e+00],\n",
       "          [-1.7470e+00, -9.9269e-01, -3.6667e+00,  ..., -6.2930e-01,\n",
       "            4.6329e-01, -2.0437e+00],\n",
       "          [ 3.3871e+00,  1.8903e-01, -3.6066e-02,  ...,  2.6472e+00,\n",
       "            1.0299e+00,  2.8266e+00]],\n",
       "\n",
       "         [[ 8.9466e-01, -1.2959e+00,  1.2044e+00,  ..., -3.4912e-01,\n",
       "           -2.9752e+00, -9.0450e-01],\n",
       "          [ 1.3989e+00,  1.4691e+00,  2.5940e+00,  ..., -1.0381e+00,\n",
       "           -3.1170e-01, -3.1089e+00],\n",
       "          [ 1.7217e-01, -4.1744e-01,  6.1825e+00,  ...,  9.5074e-01,\n",
       "           -1.7892e+00,  2.2953e+00],\n",
       "          ...,\n",
       "          [-1.5399e+00,  5.8173e-01,  3.2763e+00,  ...,  4.0914e-01,\n",
       "            1.5259e+00, -1.3484e+00],\n",
       "          [-6.2454e-01,  1.8527e-01,  2.8592e+00,  ..., -2.8442e-02,\n",
       "           -7.2541e-04,  1.2006e+00],\n",
       "          [ 1.5697e+00,  1.2694e+00, -2.0206e+00,  ...,  1.2988e+00,\n",
       "           -1.8184e+00,  1.8365e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.2298e+00, -4.3123e-01,  1.8969e+00,  ..., -8.6611e-01,\n",
       "            1.2501e+00,  3.1148e+00],\n",
       "          [-2.4078e+00, -1.2033e+00, -1.4348e+00,  ..., -4.3567e-01,\n",
       "            2.5241e+00,  1.5807e+00],\n",
       "          [-2.3631e-01, -1.4725e+00,  9.7369e-01,  ..., -6.0423e-01,\n",
       "           -2.2573e+00, -9.5005e-01],\n",
       "          ...,\n",
       "          [-8.0956e-01,  1.5565e+00,  1.4424e+00,  ..., -1.9121e+00,\n",
       "           -1.7620e+00,  8.3842e-01],\n",
       "          [-3.0191e-01,  1.1625e+00,  1.5885e+00,  ..., -2.7180e+00,\n",
       "            7.2607e-01,  5.5278e-01],\n",
       "          [-2.7277e+00, -7.3369e-01,  3.9909e+00,  ...,  1.5223e+00,\n",
       "           -4.9279e-01, -1.1963e-01]],\n",
       "\n",
       "         [[ 1.5631e+00,  3.1160e-01, -1.4413e+00,  ...,  1.7005e+00,\n",
       "            3.0796e-01, -1.2172e+00],\n",
       "          [-2.5313e+00, -1.4061e+00,  3.2967e+00,  ...,  4.0659e-01,\n",
       "            2.9010e+00, -1.1725e+00],\n",
       "          [-1.9653e+00,  3.0581e+00, -9.4067e-01,  ...,  1.3398e+00,\n",
       "            6.1033e-01,  1.0065e+00],\n",
       "          ...,\n",
       "          [-3.7374e-01,  2.8868e+00, -2.6141e-01,  ...,  8.4693e-01,\n",
       "            1.3892e-01, -1.0355e+00],\n",
       "          [ 1.2599e-01,  2.1647e+00,  2.8062e+00,  ..., -4.4028e-01,\n",
       "           -7.3323e-01, -9.0451e-01],\n",
       "          [ 1.4935e+00, -9.9172e-01,  3.8966e+00,  ...,  7.2454e-01,\n",
       "           -1.0655e+00,  1.1877e+00]],\n",
       "\n",
       "         [[ 5.8637e-01,  6.6576e-01,  3.0118e+00,  ...,  4.4548e-01,\n",
       "            6.2614e-01, -5.2357e-01],\n",
       "          [ 3.5624e-01, -7.8030e-01, -8.6868e-01,  ...,  5.9052e-01,\n",
       "            7.5537e-01, -4.4082e-01],\n",
       "          [ 2.7123e+00, -1.8718e+00,  2.2550e+00,  ..., -9.6173e-01,\n",
       "            3.7658e-01, -1.6101e+00],\n",
       "          ...,\n",
       "          [-2.1788e+00,  3.5228e+00, -5.8458e-01,  ...,  1.1217e-01,\n",
       "           -3.9637e-01, -5.5815e-02],\n",
       "          [-1.6370e+00,  1.1151e+00,  4.8705e-01,  ...,  7.7504e-01,\n",
       "           -5.0496e-02, -8.0498e-01],\n",
       "          [-9.0248e-01,  1.0204e+00, -3.7149e+00,  ...,  1.1297e+00,\n",
       "           -4.8218e-01,  7.3388e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.5475e+00,  1.6709e+00, -1.3923e+00,  ..., -5.5466e-01,\n",
       "           -1.4690e+00, -6.6920e-01],\n",
       "          [-1.2098e+00,  2.9372e+00, -3.4998e-01,  ..., -2.2676e+00,\n",
       "           -2.1711e+00,  1.1989e+00],\n",
       "          [ 1.6104e-01,  2.8594e-01, -1.0655e+00,  ..., -1.0051e+00,\n",
       "           -1.4746e+00,  1.3895e+00],\n",
       "          ...,\n",
       "          [-1.8997e-01, -1.3104e-01,  1.3266e+00,  ..., -3.0914e+00,\n",
       "           -1.5656e+00, -2.1333e+00],\n",
       "          [-1.0386e-01, -2.8043e-01,  7.9254e-01,  ..., -9.4355e-01,\n",
       "           -1.1768e+00, -2.1966e+00],\n",
       "          [ 1.0673e-01, -2.4532e+00, -3.1245e+00,  ...,  3.4332e+00,\n",
       "            5.6033e-01, -2.1052e+00]],\n",
       "\n",
       "         [[ 2.5068e-01,  1.2970e+00, -4.6286e-01,  ...,  2.3375e+00,\n",
       "            2.1146e-01, -1.6255e+00],\n",
       "          [-2.1188e+00,  3.1073e+00, -1.6144e+00,  ..., -1.0042e+00,\n",
       "            1.7484e-01, -1.8033e+00],\n",
       "          [-9.2033e-01,  1.7674e+00, -4.6787e+00,  ...,  4.1809e-01,\n",
       "            1.2442e+00,  1.6310e+00],\n",
       "          ...,\n",
       "          [ 2.4250e+00,  2.6648e-01, -2.1541e+00,  ...,  1.2168e-01,\n",
       "            9.5467e-01,  1.9954e+00],\n",
       "          [ 1.0307e+00, -1.0787e+00, -1.6461e+00,  ...,  9.9132e-02,\n",
       "            1.8021e+00, -1.7653e+00],\n",
       "          [ 3.4241e+00,  2.3431e-01,  3.7127e-02,  ...,  2.6975e+00,\n",
       "            1.0383e+00,  2.8150e+00]],\n",
       "\n",
       "         [[ 4.2247e-01,  1.1048e+00, -3.0218e+00,  ..., -2.5464e+00,\n",
       "           -5.8889e-01, -2.4769e+00],\n",
       "          [-2.8030e-01,  1.8156e-01,  5.3692e+00,  ...,  1.6322e+00,\n",
       "           -7.6586e-01,  2.8454e+00],\n",
       "          [-1.1257e+00,  2.2876e+00,  4.1801e-01,  ...,  6.6359e-01,\n",
       "           -3.0194e+00,  1.1863e+00],\n",
       "          ...,\n",
       "          [-5.0666e-01, -3.5046e-01, -1.9845e+00,  ..., -1.5297e+00,\n",
       "           -1.0571e+00, -5.3606e+00],\n",
       "          [-3.2676e-01, -2.4929e-01,  3.5070e-01,  ..., -1.7729e+00,\n",
       "            9.6452e-01, -2.3119e+00],\n",
       "          [ 1.6481e+00,  1.2154e+00, -2.0515e+00,  ...,  1.2199e+00,\n",
       "           -1.7644e+00,  1.3554e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.2772e+00, -6.2849e-01,  1.1824e+00,  ..., -3.3483e+00,\n",
       "            1.3709e+00,  6.5178e-01],\n",
       "          [-4.0466e-01, -4.8108e-01, -6.8555e-01,  ..., -1.6485e+00,\n",
       "            2.4316e+00,  2.2638e+00],\n",
       "          [-3.8393e+00, -3.3896e+00, -1.7021e-01,  ..., -1.5671e+00,\n",
       "            1.5382e+00, -3.0839e+00],\n",
       "          ...,\n",
       "          [-6.1708e-01, -2.8975e+00,  2.1813e+00,  ..., -1.3374e+00,\n",
       "           -7.3368e-01, -3.6880e+00],\n",
       "          [ 1.4617e+00,  1.2003e+00, -4.3979e-01,  ..., -2.3375e+00,\n",
       "            5.3836e-01, -5.5255e-01],\n",
       "          [-2.7943e+00, -8.1087e-01,  3.9725e+00,  ...,  1.5279e+00,\n",
       "           -4.5721e-01, -1.2866e-01]],\n",
       "\n",
       "         [[ 2.4363e+00,  2.0121e+00,  2.5925e+00,  ..., -1.6578e+00,\n",
       "            2.7408e+00,  1.8163e-01],\n",
       "          [ 2.2104e+00,  2.5827e+00,  4.0651e+00,  ..., -5.1397e-01,\n",
       "           -2.8016e+00,  1.3493e+00],\n",
       "          [ 1.8937e+00, -9.6979e-01, -1.6987e+00,  ..., -3.3980e+00,\n",
       "            3.0177e+00,  4.2367e+00],\n",
       "          ...,\n",
       "          [ 5.5263e-01, -1.9602e+00, -1.7172e+00,  ...,  9.2262e-01,\n",
       "           -9.8992e-02,  2.3505e+00],\n",
       "          [-1.7572e-01,  6.7363e-01, -6.1627e-01,  ..., -6.0282e-01,\n",
       "           -1.0659e+00, -3.7649e-01],\n",
       "          [ 1.4719e+00, -9.5598e-01,  3.9058e+00,  ...,  7.0451e-01,\n",
       "           -1.0461e+00,  1.2244e+00]],\n",
       "\n",
       "         [[ 9.1270e-01,  5.1126e-01,  2.5449e+00,  ..., -5.6950e-01,\n",
       "            2.4102e-01, -3.8155e+00],\n",
       "          [ 1.0206e+00, -3.4264e+00, -1.3897e+00,  ..., -5.3639e-01,\n",
       "           -9.8272e-01, -2.7273e+00],\n",
       "          [ 1.4663e+00, -2.5245e+00,  8.5356e-01,  ...,  9.7211e-02,\n",
       "           -9.6334e-02, -3.0041e+00],\n",
       "          ...,\n",
       "          [ 2.1465e-01,  2.0396e+00,  1.6159e+00,  ..., -1.1990e+00,\n",
       "            1.2341e+00,  7.3798e-01],\n",
       "          [-8.5833e-01, -2.2732e-01,  1.2782e+00,  ..., -4.9939e-01,\n",
       "           -2.1449e+00,  4.2127e-01],\n",
       "          [-8.2192e-01,  8.6769e-01, -3.6054e+00,  ...,  1.1325e+00,\n",
       "           -6.8074e-01,  7.5223e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-4.7470e-01,  6.9598e-01,  1.1334e+00,  ..., -1.1983e-01,\n",
       "           -3.9036e+00, -4.7096e-01],\n",
       "          [ 1.7806e+00,  1.5524e+00, -2.5429e+00,  ...,  3.7670e-01,\n",
       "           -6.4152e-01, -6.5774e-01],\n",
       "          [ 1.9695e-01,  2.9114e+00, -1.4031e+00,  ..., -3.6948e+00,\n",
       "           -1.8293e+00, -9.1101e-01],\n",
       "          ...,\n",
       "          [-1.2784e+00,  3.1934e-01, -1.6538e-01,  ...,  1.8253e+00,\n",
       "           -1.9336e+00, -1.0697e+00],\n",
       "          [-2.2931e+00, -1.1651e-01, -1.0292e+00,  ...,  1.2633e+00,\n",
       "           -3.3168e+00, -1.9110e+00],\n",
       "          [ 1.1316e-01, -2.5096e+00, -3.1315e+00,  ...,  3.5621e+00,\n",
       "            5.6798e-01, -2.1635e+00]],\n",
       "\n",
       "         [[ 1.9096e+00, -1.5510e+00,  7.2023e-02,  ...,  2.7286e-01,\n",
       "            2.1151e+00, -1.7479e+00],\n",
       "          [-1.2994e+00,  4.2222e+00, -1.3806e+00,  ..., -2.2696e+00,\n",
       "            2.6452e-01, -4.1693e+00],\n",
       "          [ 2.8182e+00,  1.8624e+00,  2.8099e+00,  ...,  3.2819e+00,\n",
       "            5.7463e-01, -3.6640e+00],\n",
       "          ...,\n",
       "          [ 6.4007e-01, -1.7344e+00,  5.2098e-01,  ...,  1.9575e+00,\n",
       "           -1.9084e-01,  9.8498e-01],\n",
       "          [ 1.9521e+00, -1.8765e-01, -1.1566e+00,  ..., -1.3359e+00,\n",
       "            2.3904e+00, -1.6446e+00],\n",
       "          [ 3.4707e+00,  2.7283e-01,  5.2596e-02,  ...,  2.6659e+00,\n",
       "            1.1455e+00,  2.8147e+00]],\n",
       "\n",
       "         [[-1.9365e+00,  3.9132e-01, -1.4369e+00,  ...,  1.2052e+00,\n",
       "           -2.3917e-01, -5.1846e-01],\n",
       "          [-5.4047e-01,  9.2048e-01, -2.1247e+00,  ...,  2.6732e+00,\n",
       "           -1.2442e+00,  3.1364e+00],\n",
       "          [ 8.3428e-01,  6.9977e-01, -4.5497e-01,  ...,  4.5236e+00,\n",
       "           -6.4355e-01,  3.2063e+00],\n",
       "          ...,\n",
       "          [-4.3524e-02, -2.3795e+00,  1.5394e+00,  ..., -1.3102e-01,\n",
       "            2.2893e+00, -7.2599e-01],\n",
       "          [-1.1092e+00, -1.7482e+00, -1.0965e+00,  ..., -2.6301e+00,\n",
       "            4.4964e-01, -1.5375e+00],\n",
       "          [ 1.5368e+00,  1.2043e+00, -2.0648e+00,  ...,  1.2600e+00,\n",
       "           -1.8003e+00,  1.2743e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 6.7343e-01, -3.1303e-01, -3.3492e-01,  ...,  3.0839e+00,\n",
       "           -8.9939e-01,  1.5628e+00],\n",
       "          [-1.1476e+00,  2.1950e+00,  1.7742e+00,  ...,  3.2112e-02,\n",
       "           -1.2411e+00, -5.7018e-01],\n",
       "          [ 1.8258e-01,  1.0329e+00,  1.8060e+00,  ..., -3.9209e-01,\n",
       "           -2.4760e+00, -8.8211e-01],\n",
       "          ...,\n",
       "          [ 1.8187e-01,  1.0829e+00,  2.3770e+00,  ..., -3.4803e-01,\n",
       "           -1.3512e+00,  5.8767e-01],\n",
       "          [ 9.5247e-01, -1.0011e+00,  1.6163e+00,  ..., -5.3264e-01,\n",
       "           -1.4198e+00,  9.1831e-01],\n",
       "          [-4.9385e-01,  2.9971e-01,  2.3553e-01,  ...,  7.1510e-02,\n",
       "            9.7425e-02,  2.4343e-01]],\n",
       "\n",
       "         [[-3.8016e-01,  5.2440e-01, -2.4819e-01,  ..., -8.1109e-01,\n",
       "            8.3772e-01,  1.8510e+00],\n",
       "          [ 2.2936e+00,  5.5992e-01, -1.7176e+00,  ...,  1.6708e+00,\n",
       "            2.3186e-01, -3.5535e-01],\n",
       "          [ 1.8459e+00,  2.1869e-01, -1.1931e+00,  ...,  4.5049e-01,\n",
       "            7.5074e-01,  9.1050e-02],\n",
       "          ...,\n",
       "          [ 2.0439e+00,  1.9161e+00, -2.5054e+00,  ...,  4.7308e-01,\n",
       "           -8.6171e-01,  1.2633e+00],\n",
       "          [ 1.0312e+00,  1.0263e-01, -9.4288e-01,  ...,  3.8233e-01,\n",
       "            6.7468e-01,  1.1266e-01],\n",
       "          [ 1.2377e+00,  1.4664e-01, -2.7671e-01,  ..., -3.0653e-01,\n",
       "            2.6291e-01,  8.7519e-01]],\n",
       "\n",
       "         [[-9.4157e-01, -6.4248e-02, -1.8584e-01,  ...,  1.4372e+00,\n",
       "           -4.7730e-01,  4.5582e-02],\n",
       "          [ 2.9401e-01,  2.7586e+00, -9.1875e-01,  ...,  1.4073e+00,\n",
       "           -9.0955e-01, -1.4287e+00],\n",
       "          [-1.5852e-01, -2.0797e-01, -1.6630e+00,  ...,  1.5778e+00,\n",
       "           -2.1911e+00, -2.3477e-01],\n",
       "          ...,\n",
       "          [ 1.1924e+00,  8.6493e-01, -1.3482e+00,  ...,  2.0486e+00,\n",
       "           -2.3518e+00, -2.3889e-01],\n",
       "          [ 7.0720e-01,  1.2649e-01, -9.1544e-02,  ...,  7.8519e-01,\n",
       "           -1.4810e+00, -6.2359e-01],\n",
       "          [ 1.9156e-01,  5.5493e-01, -1.5672e-01,  ...,  1.0761e+00,\n",
       "            1.2795e+00,  2.9820e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-7.2778e-01, -1.0648e+00, -2.5947e+00,  ...,  6.4872e-02,\n",
       "           -8.8260e-01,  1.6908e+00],\n",
       "          [-1.2446e+00, -2.0421e+00, -9.6885e-01,  ...,  3.4387e+00,\n",
       "           -4.0072e-01,  8.5198e-01],\n",
       "          [-1.9331e+00,  8.9582e-02, -1.4050e+00,  ...,  1.4867e+00,\n",
       "           -1.7098e+00, -9.3491e-02],\n",
       "          ...,\n",
       "          [ 1.3328e+00, -1.4295e+00, -1.1214e+00,  ...,  1.5529e+00,\n",
       "            3.7557e-01,  1.0537e+00],\n",
       "          [ 1.0569e+00, -6.4826e-01, -7.2836e-01,  ...,  1.1311e-01,\n",
       "            7.2231e-01,  9.3609e-01],\n",
       "          [ 1.0041e-01, -4.1503e-01, -4.2553e-01,  ..., -1.8100e-02,\n",
       "           -2.4121e-01,  5.1562e-01]],\n",
       "\n",
       "         [[ 2.0761e-01, -1.6120e+00, -1.5240e-01,  ...,  1.9550e-01,\n",
       "            4.4150e-01, -5.6047e-01],\n",
       "          [-9.7466e-02,  6.1491e-01, -4.3336e-01,  ..., -1.2673e-01,\n",
       "            2.7613e+00, -2.7303e-01],\n",
       "          [-1.6474e+00,  9.9123e-01,  1.1684e+00,  ..., -1.0377e+00,\n",
       "            1.4127e+00,  4.8426e-01],\n",
       "          ...,\n",
       "          [-4.7770e-01, -4.9826e-01, -6.0742e-02,  ..., -1.6697e+00,\n",
       "            1.1738e+00,  5.7650e-01],\n",
       "          [-4.5834e-01, -1.3396e-01, -9.6062e-02,  ..., -3.5937e-01,\n",
       "            2.9862e-01,  1.3705e+00],\n",
       "          [ 5.8377e-01,  3.5587e-01,  6.6314e-02,  ...,  5.6760e-02,\n",
       "            5.4712e-02, -1.7735e-01]],\n",
       "\n",
       "         [[-2.8459e-01, -1.4886e+00, -5.2668e-01,  ...,  1.2604e+00,\n",
       "           -5.1129e-01,  4.0525e-01],\n",
       "          [ 1.2541e-02,  5.2177e-01, -1.2454e+00,  ...,  4.7700e-01,\n",
       "            2.2776e+00,  1.2164e+00],\n",
       "          [-8.7862e-01, -9.8957e-01, -9.3768e-01,  ..., -6.7176e-01,\n",
       "            7.1942e-01,  6.9871e-01],\n",
       "          ...,\n",
       "          [-1.3553e+00,  8.5923e-01, -6.9523e-01,  ..., -9.9951e-01,\n",
       "            5.5646e-01,  9.2985e-01],\n",
       "          [-9.4786e-01, -2.8235e-01, -1.7851e+00,  ..., -2.3386e-01,\n",
       "            2.4112e-01,  3.8405e-01],\n",
       "          [-3.9683e-01, -2.9109e-01,  1.7786e-01,  ...,  1.0915e-01,\n",
       "            2.4873e-01, -7.5593e-02]]],\n",
       "\n",
       "\n",
       "        [[[ 4.3850e-01, -1.9369e-01,  5.8258e-01,  ...,  3.7707e-01,\n",
       "           -1.9838e+00,  4.5672e-01],\n",
       "          [ 9.0951e-01,  6.5982e-01,  1.5785e+00,  ..., -8.2153e-01,\n",
       "           -2.4455e+00, -7.6205e-01],\n",
       "          [ 1.7260e+00,  5.9255e-01,  1.6876e+00,  ...,  9.7238e-01,\n",
       "           -7.6730e-01, -3.0008e-01],\n",
       "          ...,\n",
       "          [ 1.1142e+00, -6.2468e-01,  1.9015e-01,  ...,  6.4655e-01,\n",
       "           -1.3817e+00,  1.3655e+00],\n",
       "          [ 1.4699e+00, -1.8310e-01,  1.7254e+00,  ..., -2.4218e-01,\n",
       "           -7.9474e-01,  8.9891e-01],\n",
       "          [-5.0611e-01,  2.8019e-01,  2.4936e-01,  ...,  2.1494e-02,\n",
       "            1.3235e-01,  2.0331e-01]],\n",
       "\n",
       "         [[ 4.6319e-01,  1.1393e+00, -1.9356e+00,  ...,  3.4179e+00,\n",
       "           -1.0718e+00, -9.2718e-01],\n",
       "          [ 1.5360e+00,  8.3947e-01, -1.5784e+00,  ..., -2.6225e-01,\n",
       "            5.3592e-01,  6.7721e-02],\n",
       "          [ 1.0533e+00,  2.4156e+00,  1.4594e+00,  ..., -7.7767e-01,\n",
       "            6.9344e-01, -1.2758e+00],\n",
       "          ...,\n",
       "          [ 5.7717e-02,  2.5869e-02,  3.7940e-01,  ..., -7.1370e-01,\n",
       "            9.7582e-01,  8.2273e-01],\n",
       "          [ 1.1846e+00,  4.0441e-01, -1.3129e+00,  ...,  1.0154e+00,\n",
       "            5.2640e-01,  5.2577e-01],\n",
       "          [ 1.2402e+00,  1.1088e-01, -2.8517e-01,  ..., -2.7891e-01,\n",
       "            2.5632e-01,  8.6709e-01]],\n",
       "\n",
       "         [[ 7.6022e-01,  4.2622e-01, -2.2682e+00,  ...,  2.0863e+00,\n",
       "           -2.5428e+00, -7.3718e-01],\n",
       "          [ 8.6999e-01,  9.1798e-01, -2.5222e+00,  ...,  1.2983e+00,\n",
       "           -3.3988e+00,  6.1353e-01],\n",
       "          [ 2.0543e+00,  1.5230e+00, -6.1752e-01,  ...,  4.3547e-01,\n",
       "           -1.9886e+00, -1.7701e-01],\n",
       "          ...,\n",
       "          [-3.8536e-01, -2.5717e-01, -2.1034e+00,  ...,  1.7195e+00,\n",
       "           -5.0930e+00, -3.9360e-01],\n",
       "          [ 2.0910e-01,  1.8992e-01, -3.7391e-01,  ...,  3.0866e+00,\n",
       "           -9.8222e-01, -9.0276e-01],\n",
       "          [ 1.9249e-01,  5.3588e-01, -1.8110e-01,  ...,  1.1205e+00,\n",
       "            1.2370e+00,  2.9460e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-9.8659e-01,  4.3449e-01, -2.1158e+00,  ...,  8.4263e-01,\n",
       "           -2.0272e+00, -2.4467e-01],\n",
       "          [-2.3043e+00,  8.9602e-01, -2.0326e+00,  ...,  1.1919e+00,\n",
       "           -1.7562e+00, -9.3214e-01],\n",
       "          [-7.7973e-01,  7.1407e-02, -3.0182e+00,  ..., -8.0790e-01,\n",
       "            2.1782e-01,  9.5648e-02],\n",
       "          ...,\n",
       "          [-5.1611e-01, -2.0803e+00, -3.3604e-01,  ..., -3.0771e-02,\n",
       "            7.9693e-01,  1.4821e-01],\n",
       "          [ 2.3975e-01, -1.0838e+00, -1.4746e+00,  ...,  3.1513e-01,\n",
       "           -8.3814e-02,  1.6133e+00],\n",
       "          [ 6.8200e-02, -3.9917e-01, -3.8468e-01,  ..., -3.4250e-02,\n",
       "           -2.2989e-01,  5.6429e-01]],\n",
       "\n",
       "         [[-1.4583e-01,  1.4043e+00, -1.5267e+00,  ..., -4.3496e-01,\n",
       "            4.4240e+00,  1.6869e+00],\n",
       "          [-1.2149e+00,  1.3389e+00,  9.6721e-01,  ..., -7.1642e-01,\n",
       "            1.3300e-01,  4.8887e-01],\n",
       "          [-2.0897e-01, -4.6083e-01,  1.5726e+00,  ...,  2.0977e+00,\n",
       "            3.8646e-01,  7.5748e-01],\n",
       "          ...,\n",
       "          [ 1.3948e-02, -3.2666e-01, -5.4247e-01,  ..., -4.8025e-01,\n",
       "           -4.0355e-01,  1.4240e+00],\n",
       "          [-6.5424e-01,  5.2443e-01, -1.2223e+00,  ..., -2.2098e-02,\n",
       "           -3.6109e-01,  9.8641e-01],\n",
       "          [ 6.2623e-01,  3.3837e-01,  6.4005e-02,  ...,  8.7061e-02,\n",
       "            2.6267e-02, -1.8106e-01]],\n",
       "\n",
       "         [[ 3.2062e-01,  9.1272e-01, -3.0889e+00,  ...,  3.8562e-01,\n",
       "            3.1526e+00,  3.1159e-01],\n",
       "          [ 5.7628e-01,  5.1767e-01, -6.5063e-01,  ..., -1.4600e+00,\n",
       "            2.4592e-01,  8.1445e-01],\n",
       "          [ 1.1154e+00,  1.6390e-01, -7.2309e-01,  ...,  6.8528e-01,\n",
       "           -1.5988e+00,  1.1794e+00],\n",
       "          ...,\n",
       "          [ 5.7190e-01,  3.9165e-02, -3.5900e-01,  ...,  5.5652e-01,\n",
       "            2.0864e+00,  1.3024e+00],\n",
       "          [-7.9524e-01,  8.7946e-01, -5.5690e-01,  ..., -3.6863e-01,\n",
       "            1.1528e+00,  1.4568e-01],\n",
       "          [-3.8028e-01, -2.9792e-01,  1.8081e-01,  ...,  1.1829e-01,\n",
       "            2.7101e-01, -8.6419e-02]]],\n",
       "\n",
       "\n",
       "        [[[ 7.8299e-02, -7.1487e-01,  9.6486e-01,  ...,  1.1185e+00,\n",
       "           -1.4567e+00,  3.0667e-01],\n",
       "          [-1.7099e+00, -8.2657e-01,  2.3102e+00,  ...,  7.1563e-02,\n",
       "           -2.4158e+00,  2.7831e-01],\n",
       "          [-3.4421e-01, -3.9709e-01,  5.3122e-01,  ..., -1.3040e+00,\n",
       "           -7.3066e-01, -3.7665e-01],\n",
       "          ...,\n",
       "          [-6.8858e-01, -6.9589e-01,  3.2946e-01,  ..., -1.1819e+00,\n",
       "           -4.1751e-01,  6.6952e-01],\n",
       "          [ 1.2947e-02,  3.0843e-01,  2.7419e+00,  ..., -3.6728e-01,\n",
       "           -2.9587e-01,  6.1692e-01],\n",
       "          [-5.0072e-01,  2.7444e-01,  2.6884e-01,  ...,  6.2994e-02,\n",
       "            9.9136e-02,  2.2841e-01]],\n",
       "\n",
       "         [[-8.0826e-01,  7.4647e-02,  2.7868e-01,  ...,  1.5579e+00,\n",
       "            8.3106e-01,  1.7972e+00],\n",
       "          [ 3.3112e+00,  1.6377e+00, -7.1252e-01,  ...,  1.4829e+00,\n",
       "           -1.3182e-01,  1.7842e+00],\n",
       "          [ 5.7545e-01, -1.0589e+00, -4.0698e-01,  ...,  1.3133e-01,\n",
       "            1.0987e+00,  1.3191e+00],\n",
       "          ...,\n",
       "          [ 8.9991e-01, -1.0081e-02, -2.4392e+00,  ...,  7.3747e-01,\n",
       "           -2.3103e-01,  2.0298e+00],\n",
       "          [ 7.9144e-01,  4.1419e-01, -1.0214e+00,  ...,  2.1814e-01,\n",
       "           -2.4065e-01,  2.0814e+00],\n",
       "          [ 1.2349e+00,  1.2165e-01, -2.6118e-01,  ..., -2.8476e-01,\n",
       "            2.7698e-01,  8.2314e-01]],\n",
       "\n",
       "         [[ 6.4168e-01, -1.8190e-01, -1.3160e+00,  ..., -5.1758e-01,\n",
       "           -1.0541e+00, -1.3403e+00],\n",
       "          [ 2.0054e+00,  8.1638e-01, -1.3513e+00,  ...,  1.8733e-01,\n",
       "           -3.0403e-01,  1.7877e+00],\n",
       "          [ 1.5636e+00,  1.4012e+00, -4.9431e-01,  ..., -5.9252e-01,\n",
       "            1.2765e+00,  2.2295e+00],\n",
       "          ...,\n",
       "          [ 1.1778e+00, -2.2415e-01, -2.0726e+00,  ...,  5.0763e-01,\n",
       "            2.9860e-01,  5.8148e-01],\n",
       "          [ 2.7112e-01, -1.2321e-01, -5.8349e-01,  ...,  2.0930e+00,\n",
       "           -4.8367e-01,  1.6923e-01],\n",
       "          [ 2.0137e-01,  5.4208e-01, -1.7969e-01,  ...,  1.1009e+00,\n",
       "            1.2674e+00,  2.9456e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-4.6553e-01, -1.5496e+00, -8.3646e-01,  ..., -1.2605e-01,\n",
       "           -9.1236e-01, -1.0267e+00],\n",
       "          [-4.5754e-01, -8.6144e-01, -1.0189e+00,  ...,  4.5751e-02,\n",
       "            6.0055e-01,  2.0493e-01],\n",
       "          [-9.5411e-01,  2.1369e-01, -1.4703e+00,  ...,  4.0652e-01,\n",
       "            1.4254e+00, -1.8129e+00],\n",
       "          ...,\n",
       "          [ 9.7233e-04, -2.0754e-01,  4.8897e-01,  ...,  6.2282e-01,\n",
       "           -1.1374e+00,  1.5626e+00],\n",
       "          [ 1.9115e+00, -6.1017e-01, -1.2243e+00,  ...,  5.9162e-01,\n",
       "           -7.8698e-01,  6.1281e-01],\n",
       "          [ 8.0911e-02, -3.9896e-01, -4.0392e-01,  ..., -1.8819e-02,\n",
       "           -2.3600e-01,  4.8352e-01]],\n",
       "\n",
       "         [[-1.2239e-01, -6.4944e-01, -5.4235e-01,  ..., -1.4470e+00,\n",
       "            2.0443e+00,  3.6390e+00],\n",
       "          [ 1.4242e-01,  7.9762e-01,  1.7006e+00,  ...,  3.5647e-01,\n",
       "            5.0444e-02, -5.9867e-02],\n",
       "          [-6.5275e-01, -6.4005e-01,  1.8436e+00,  ...,  1.6326e+00,\n",
       "           -3.0510e-01, -2.3990e+00],\n",
       "          ...,\n",
       "          [-2.7749e-01, -9.0164e-04,  1.6351e+00,  ..., -3.1856e-01,\n",
       "            5.5564e-01, -1.1209e+00],\n",
       "          [-6.3164e-01,  3.6362e-02,  1.2482e-02,  ...,  5.6626e-01,\n",
       "           -7.7724e-01,  1.2429e+00],\n",
       "          [ 6.1140e-01,  3.4619e-01,  7.5600e-02,  ...,  6.4677e-02,\n",
       "            3.0913e-02, -1.9349e-01]],\n",
       "\n",
       "         [[ 1.9255e+00,  1.3367e+00, -1.7847e+00,  ...,  1.8512e+00,\n",
       "            8.1166e-01,  5.8044e-02],\n",
       "          [ 7.8968e-01, -1.5312e-01,  7.2520e-01,  ..., -1.4983e-01,\n",
       "           -6.9544e-01, -2.9593e-01],\n",
       "          [ 2.4190e+00,  3.4548e-01,  1.2258e+00,  ...,  9.3930e-01,\n",
       "            7.7061e-02, -4.0930e-01],\n",
       "          ...,\n",
       "          [-1.1693e+00,  7.6643e-01, -1.9625e+00,  ...,  1.7556e+00,\n",
       "            3.7383e-01, -6.3633e-01],\n",
       "          [-2.3768e-01, -3.2747e-01, -1.6496e-01,  ..., -9.8126e-02,\n",
       "            9.4097e-01, -1.5197e-01],\n",
       "          [-3.7748e-01, -2.9530e-01,  2.0504e-01,  ...,  1.4198e-01,\n",
       "            2.1146e-01, -9.0703e-02]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[ 0.2294, -0.2760, -0.7463,  ..., -1.0828,  1.5068,  0.4261],\n",
       "          [ 0.3001, -0.0767, -0.4033,  ..., -0.2100,  0.9318, -0.0771],\n",
       "          [ 0.1450,  0.0822, -0.5826,  ..., -0.1825,  0.6323, -0.1721],\n",
       "          ...,\n",
       "          [-0.1299,  0.4199, -0.1838,  ..., -0.3969,  0.5692, -0.2822],\n",
       "          [ 0.0366,  0.0403, -0.2232,  ..., -0.4722,  0.9389, -0.3864],\n",
       "          [ 1.0195,  0.4270, -0.5041,  ..., -0.8036,  1.4347, -0.9309]],\n",
       "\n",
       "         [[-0.8521,  0.9341, -1.2829,  ...,  1.1964, -0.3201,  1.2441],\n",
       "          [-0.4820,  0.0022, -0.1552,  ...,  0.3832,  0.1213,  0.3472],\n",
       "          [-0.2491, -0.1771,  0.1585,  ...,  0.2661, -0.1165,  0.1296],\n",
       "          ...,\n",
       "          [-0.0186,  0.1320, -0.4299,  ...,  0.7510, -0.3710,  0.2714],\n",
       "          [-0.0703,  0.0457, -0.4097,  ...,  0.6669, -0.1385,  0.2533],\n",
       "          [-0.2120, -0.1051, -0.3422,  ...,  0.8026,  0.1478,  0.3910]],\n",
       "\n",
       "         [[-0.3023, -0.2417,  1.1354,  ..., -0.1449,  0.2306, -0.5451],\n",
       "          [ 0.0161,  0.1914,  0.5205,  ..., -0.1108, -0.6139, -0.2965],\n",
       "          [-0.0297,  0.2259,  0.2569,  ..., -0.1649, -0.5612, -0.1714],\n",
       "          ...,\n",
       "          [-0.1568, -0.1228,  0.1797,  ...,  0.2065, -0.3837, -0.1923],\n",
       "          [ 0.0599, -0.3697,  0.2468,  ...,  0.0243, -0.3441, -0.2444],\n",
       "          [-0.2772, -0.1548,  0.1260,  ...,  0.4598, -0.1702, -0.3149]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 0.3894,  0.1548, -0.6278,  ...,  0.7737,  0.0155, -0.3610],\n",
       "          [ 0.0037, -0.3013,  0.2059,  ..., -0.1951,  0.7223,  0.2012],\n",
       "          [-0.2549,  0.0784,  0.1957,  ..., -0.0157,  0.4713,  0.1584],\n",
       "          ...,\n",
       "          [ 0.0650,  0.0603, -0.2952,  ...,  0.3540, -0.2119, -0.1004],\n",
       "          [ 0.0073,  0.0464, -0.1279,  ...,  0.3295, -0.0942,  0.2656],\n",
       "          [ 0.3142, -0.0183, -0.5729,  ...,  0.4033,  0.2000, -0.0493]],\n",
       "\n",
       "         [[-0.3919,  0.4888,  0.4597,  ...,  0.0322,  0.6630, -0.6963],\n",
       "          [ 0.3912, -0.0871,  0.1234,  ..., -0.2246,  0.6298, -0.5228],\n",
       "          [ 0.1171, -0.0453,  0.0855,  ...,  0.1536,  0.1279,  0.0808],\n",
       "          ...,\n",
       "          [-0.1052,  0.0141,  0.0809,  ..., -0.0432,  0.6202, -0.4920],\n",
       "          [ 0.2104,  0.1335,  0.1209,  ...,  0.0088,  0.4208, -0.2534],\n",
       "          [ 0.3675,  0.0050,  0.2840,  ..., -0.1479,  0.1077, -0.3657]],\n",
       "\n",
       "         [[ 0.4026,  0.5935,  0.5100,  ..., -0.2750,  0.0440, -0.2646],\n",
       "          [ 0.2504,  0.2454, -0.4533,  ..., -0.2858, -0.5924,  0.2615],\n",
       "          [ 0.1410,  0.4275, -0.1800,  ..., -0.2709, -0.2182,  0.3256],\n",
       "          ...,\n",
       "          [ 0.2360, -0.0539, -0.0178,  ...,  0.0508, -0.0959,  0.1088],\n",
       "          [-0.2360,  0.1167, -0.2045,  ..., -0.0551,  0.1079,  0.0433],\n",
       "          [ 0.0903, -0.1196, -0.3241,  ..., -0.3243,  0.2483,  0.9534]]],\n",
       "\n",
       "\n",
       "        [[[ 0.2155, -0.2691, -0.7398,  ..., -1.0755,  1.5213,  0.4424],\n",
       "          [ 0.7298,  0.2619, -0.5430,  ..., -0.5437,  1.1769,  0.2453],\n",
       "          [ 0.1344,  0.0213, -0.2860,  ..., -0.4081,  0.9800, -0.2598],\n",
       "          ...,\n",
       "          [ 0.2411,  0.4036, -0.4713,  ..., -0.0583,  0.7585, -0.4342],\n",
       "          [-0.2189,  0.2233, -0.7855,  ..., -0.5705,  0.9951, -0.1954],\n",
       "          [ 0.5473,  0.1455, -0.5496,  ..., -0.4115,  0.9247, -0.3275]],\n",
       "\n",
       "         [[-0.8504,  0.9426, -1.2806,  ...,  1.2185, -0.3764,  1.2380],\n",
       "          [-0.0936, -0.0968, -0.7743,  ...,  1.0394, -0.3275,  0.1657],\n",
       "          [-0.1685,  0.0643, -0.4110,  ...,  0.7855, -0.3009,  0.3065],\n",
       "          ...,\n",
       "          [-0.2890, -0.2987, -0.6465,  ...,  1.1177, -0.3987,  0.2845],\n",
       "          [-0.2550,  0.0590, -0.3444,  ...,  0.7392, -0.0581,  0.3525],\n",
       "          [-0.2503,  0.0187, -0.4065,  ...,  0.8272,  0.1507,  0.1226]],\n",
       "\n",
       "         [[-0.2799, -0.2230,  1.1172,  ..., -0.2040,  0.2392, -0.5826],\n",
       "          [-0.0595,  0.2572,  0.5513,  ..., -0.3836, -0.3149,  0.1300],\n",
       "          [ 0.0046,  0.1501,  0.1862,  ..., -0.2597, -0.3885, -0.0879],\n",
       "          ...,\n",
       "          [-0.0026, -0.1392,  0.4748,  ..., -0.1891, -0.4595, -0.2375],\n",
       "          [ 0.1749,  0.2144,  0.3940,  ...,  0.0346, -0.0882, -0.3698],\n",
       "          [ 0.1495,  0.1418,  0.4119,  ...,  0.1699, -0.1838, -0.3793]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 0.3796,  0.1521, -0.6576,  ...,  0.7820, -0.0058, -0.4053],\n",
       "          [ 0.0212, -0.0417,  0.1260,  ...,  0.1497,  0.0395,  0.1568],\n",
       "          [ 0.0416,  0.0343,  0.0734,  ...,  0.4416,  0.0827,  0.1111],\n",
       "          ...,\n",
       "          [-0.1222, -0.0737, -0.1091,  ..., -0.0172,  0.0656, -0.3137],\n",
       "          [ 0.1882,  0.0374, -0.2299,  ...,  0.3369, -0.2372, -0.1062],\n",
       "          [ 0.0890,  0.0884, -0.4579,  ...,  0.4430,  0.2535, -0.2011]],\n",
       "\n",
       "         [[-0.3937,  0.5291,  0.4334,  ...,  0.0292,  0.6770, -0.6755],\n",
       "          [ 0.1354, -0.0056,  0.1072,  ..., -0.2189,  0.5716, -0.7463],\n",
       "          [-0.1840, -0.0981,  0.0458,  ...,  0.1142,  0.4936, -0.3189],\n",
       "          ...,\n",
       "          [-0.2388,  0.1388,  0.1601,  ...,  0.1231,  0.1727, -0.3016],\n",
       "          [ 0.0628,  0.3048,  0.2454,  ...,  0.0986,  0.4407, -0.1891],\n",
       "          [ 0.2000,  0.0830,  0.1617,  ...,  0.0573,  0.3692, -0.3090]],\n",
       "\n",
       "         [[ 0.4046,  0.5538,  0.4888,  ..., -0.2665,  0.0730, -0.2640],\n",
       "          [ 0.3836, -0.3269, -0.4862,  ..., -0.1493,  0.1538,  0.4744],\n",
       "          [ 0.0553, -0.1064, -0.2571,  ..., -0.1810, -0.1783,  0.2058],\n",
       "          ...,\n",
       "          [ 0.1129, -0.1254, -0.2339,  ..., -0.2981, -0.2155,  0.1795],\n",
       "          [ 0.0902,  0.2215,  0.0094,  ..., -0.0610, -0.1278,  0.0479],\n",
       "          [ 0.3010,  0.0725, -0.4115,  ..., -0.0225,  0.0537,  0.2760]]],\n",
       "\n",
       "\n",
       "        [[[ 0.2212, -0.2565, -0.7467,  ..., -1.0784,  1.5160,  0.4394],\n",
       "          [ 0.1116,  0.0330, -0.5063,  ..., -0.7798,  1.0254, -0.0275],\n",
       "          [-0.0379, -0.0979, -0.6206,  ..., -0.7109,  0.8655, -0.1185],\n",
       "          ...,\n",
       "          [ 0.0288,  0.2769, -0.2643,  ..., -0.7532,  1.0601, -0.4587],\n",
       "          [ 0.2039,  0.0284, -0.4904,  ..., -0.6632,  0.9783, -0.2793],\n",
       "          [ 0.8871,  0.0999, -0.2704,  ..., -0.6548,  1.1937, -0.9454]],\n",
       "\n",
       "         [[-0.8626,  0.9040, -1.2611,  ...,  1.2079, -0.3363,  1.2229],\n",
       "          [-0.2990,  0.3925, -0.8695,  ...,  1.1143, -0.2257,  0.3804],\n",
       "          [-0.2677,  0.3046, -0.5927,  ...,  0.9475,  0.1442,  0.1221],\n",
       "          ...,\n",
       "          [-0.5220,  0.2003, -0.5013,  ...,  0.8184,  0.0362, -0.0277],\n",
       "          [-0.2614,  0.0911, -0.4367,  ...,  0.9884, -0.0339,  0.0525],\n",
       "          [-0.4355, -0.0964, -0.8393,  ...,  0.7280,  0.1884,  0.0799]],\n",
       "\n",
       "         [[-0.2852, -0.2400,  1.1070,  ..., -0.1619,  0.2118, -0.5532],\n",
       "          [-0.0301, -0.0484,  0.4467,  ..., -0.2579, -0.2437, -0.1954],\n",
       "          [-0.0924,  0.1045,  0.5388,  ..., -0.2205, -0.1630, -0.4181],\n",
       "          ...,\n",
       "          [-0.0775, -0.0837,  0.0774,  ...,  0.3712, -0.5155, -0.0487],\n",
       "          [ 0.2474,  0.1588,  0.4848,  ..., -0.0267, -0.4895, -0.2473],\n",
       "          [ 0.0215, -0.1107,  0.2490,  ...,  0.4515, -0.3261, -0.2394]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 0.4004,  0.1359, -0.6350,  ...,  0.7448,  0.0250, -0.3962],\n",
       "          [ 0.3576, -0.0325, -0.2069,  ...,  0.3415,  0.2701, -0.0870],\n",
       "          [ 0.1132,  0.1310, -0.3104,  ...,  0.2691,  0.1177, -0.1694],\n",
       "          ...,\n",
       "          [ 0.3536, -0.1792, -0.3910,  ...,  0.3789,  0.2126, -0.0326],\n",
       "          [ 0.2811, -0.0664, -0.2944,  ...,  0.3173,  0.0346, -0.1672],\n",
       "          [ 0.1438, -0.0270, -0.3579,  ...,  0.4120,  0.3601,  0.1603]],\n",
       "\n",
       "         [[-0.3822,  0.5427,  0.4436,  ...,  0.0257,  0.6602, -0.6866],\n",
       "          [-0.0863,  0.3262,  0.5426,  ..., -0.0658,  0.3831, -0.4184],\n",
       "          [ 0.1624,  0.2247,  0.1157,  ...,  0.0903,  0.4217, -0.2457],\n",
       "          ...,\n",
       "          [ 0.1393, -0.0054,  0.2723,  ..., -0.1126,  0.2831, -0.5301],\n",
       "          [ 0.0517,  0.1827,  0.2349,  ...,  0.1059,  0.5629, -0.2515],\n",
       "          [ 0.1891,  0.0813,  0.2666,  ..., -0.1519,  0.2615, -0.6439]],\n",
       "\n",
       "         [[ 0.4076,  0.5633,  0.4852,  ..., -0.2655,  0.0546, -0.2739],\n",
       "          [ 0.1118,  0.3783, -0.1348,  ..., -0.3677, -0.0380, -0.1977],\n",
       "          [-0.0415,  0.2046, -0.0725,  ..., -0.3824, -0.0949, -0.2797],\n",
       "          ...,\n",
       "          [ 0.2131,  0.1272, -0.0534,  ...,  0.0238, -0.1930, -0.2259],\n",
       "          [ 0.1799,  0.3193,  0.0642,  ...,  0.0346, -0.1782,  0.0457],\n",
       "          [-0.1006, -0.0546, -0.2914,  ..., -0.2732,  0.0621,  0.5128]]]],\n",
       "       grad_fn=<TransposeBackward0>), tensor([[[[ 1.7218e-01, -1.7093e-01, -3.7944e-01,  ..., -1.3119e-01,\n",
       "           -4.1105e-02, -5.5904e-02],\n",
       "          [ 3.3360e-01, -5.6309e-02,  3.2222e-01,  ...,  1.7957e-01,\n",
       "           -3.9907e-03,  3.1606e-01],\n",
       "          [ 3.3994e-01, -6.4852e-02,  4.3534e-01,  ...,  3.0501e-01,\n",
       "           -1.9301e-01,  9.0302e-02],\n",
       "          ...,\n",
       "          [ 3.9483e-01,  3.1554e-02,  3.0945e-01,  ..., -1.8730e-01,\n",
       "           -1.0354e-01, -2.2137e-01],\n",
       "          [ 3.7689e-01, -1.9093e-01,  2.1487e-01,  ..., -1.5791e-01,\n",
       "            1.0717e-02, -5.9608e-02],\n",
       "          [-2.2935e-01,  1.2517e-01,  3.6149e-01,  ...,  2.6885e-01,\n",
       "           -3.3676e-01,  4.7676e-02]],\n",
       "\n",
       "         [[-4.4275e-01,  2.2911e-01,  4.0775e-01,  ..., -3.9183e-01,\n",
       "            5.6122e-01, -9.8058e-01],\n",
       "          [-1.3902e-01,  2.7166e-02,  5.7405e-01,  ...,  2.3324e-01,\n",
       "            2.2972e-02, -1.8667e-01],\n",
       "          [-1.6454e-01, -5.7080e-02,  6.0567e-01,  ...,  7.8923e-03,\n",
       "            5.4751e-03, -4.2715e-01],\n",
       "          ...,\n",
       "          [-4.2898e-01, -3.1893e-02,  8.7380e-02,  ..., -1.9496e-01,\n",
       "            2.3213e-01, -2.7396e-01],\n",
       "          [-1.9946e-01, -5.5414e-02,  4.3141e-01,  ...,  3.7470e-02,\n",
       "            2.9297e-01, -5.1061e-01],\n",
       "          [-8.5624e-02,  1.2309e-01,  5.2083e-01,  ...,  4.9289e-02,\n",
       "            3.8541e-03, -7.6740e-01]],\n",
       "\n",
       "         [[ 1.8753e-01, -2.6352e-01,  5.8248e-01,  ..., -8.9470e-02,\n",
       "            1.9116e-01, -1.3770e-01],\n",
       "          [ 4.5920e-02, -2.1357e-01,  1.0952e+00,  ...,  9.4108e-03,\n",
       "            8.8258e-04, -1.2684e-01],\n",
       "          [-3.0709e-02, -4.1855e-02,  8.2991e-01,  ..., -1.3761e-01,\n",
       "            2.8859e-01, -6.1141e-02],\n",
       "          ...,\n",
       "          [ 8.3221e-02,  4.0429e-05,  6.8835e-01,  ...,  4.5305e-02,\n",
       "           -9.4366e-02,  4.0041e-02],\n",
       "          [ 5.9352e-02, -7.8941e-03,  5.6785e-01,  ..., -5.1271e-02,\n",
       "            7.3186e-02,  1.8401e-01],\n",
       "          [ 4.6888e-01, -3.3869e-01,  6.6258e-01,  ..., -3.3677e-01,\n",
       "            1.5591e-01, -4.6401e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.2008e-01, -5.6859e-02,  8.3879e-02,  ...,  2.8729e-01,\n",
       "           -1.4246e-01,  8.0496e-01],\n",
       "          [-2.8778e-01, -3.3279e-01, -6.7435e-03,  ...,  3.6940e-01,\n",
       "            1.9278e-01,  5.0163e-01],\n",
       "          [-1.7444e-01, -7.5730e-02,  7.3695e-02,  ...,  3.3983e-01,\n",
       "            1.1597e-03,  6.1490e-01],\n",
       "          ...,\n",
       "          [ 6.4804e-02, -2.7500e-01, -8.8951e-02,  ...,  3.8154e-01,\n",
       "            7.5454e-02,  5.7810e-01],\n",
       "          [-1.4503e-01, -2.5465e-01,  1.7297e-01,  ...,  2.5612e-01,\n",
       "            9.2934e-02,  4.6258e-01],\n",
       "          [-1.8753e-01, -7.6342e-01,  2.6607e-01,  ...,  4.5084e-01,\n",
       "           -3.8817e-01,  6.9587e-01]],\n",
       "\n",
       "         [[ 1.9671e-02,  2.3634e-01, -1.1913e-01,  ..., -8.7448e-01,\n",
       "           -3.7516e-01,  4.2265e-01],\n",
       "          [-5.7665e-02,  4.0831e-01,  8.5390e-02,  ..., -1.2910e+00,\n",
       "           -1.3277e-01,  5.1256e-01],\n",
       "          [-2.4317e-02,  3.9758e-01,  1.2567e-01,  ..., -1.0244e+00,\n",
       "           -4.0848e-01,  4.2281e-01],\n",
       "          ...,\n",
       "          [ 1.1446e-01, -5.6162e-02, -1.4414e-01,  ..., -5.3764e-01,\n",
       "           -1.7770e-01,  1.7611e-01],\n",
       "          [-1.1945e-01,  3.5069e-01, -4.1356e-01,  ..., -8.4168e-01,\n",
       "           -4.4923e-01,  2.5792e-01],\n",
       "          [-9.4587e-02,  5.8899e-01, -5.6831e-01,  ..., -4.8077e-01,\n",
       "            2.6290e-01, -2.4653e-01]],\n",
       "\n",
       "         [[-1.4134e-02, -4.7636e-01, -1.2050e-01,  ...,  1.8340e-02,\n",
       "           -2.0916e-01, -2.7487e-01],\n",
       "          [-2.7112e-01, -2.1559e-02, -1.7772e-01,  ...,  5.6140e-01,\n",
       "           -1.8865e-01,  6.3558e-01],\n",
       "          [-8.7744e-02,  2.3975e-02, -6.9783e-02,  ...,  4.2441e-01,\n",
       "           -1.7798e-01,  4.2125e-01],\n",
       "          ...,\n",
       "          [-7.0180e-02,  2.5714e-01, -1.0360e-01,  ...,  8.4882e-02,\n",
       "           -2.9080e-01,  1.8577e-01],\n",
       "          [ 4.4634e-02,  1.5471e-01,  4.1543e-02,  ..., -1.5796e-01,\n",
       "           -5.0948e-01,  1.8433e-01],\n",
       "          [ 4.3456e-01, -1.1178e-01,  4.4397e-02,  ...,  4.8490e-01,\n",
       "           -1.5663e-01,  4.6810e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.5120e-01, -1.8866e-01, -3.8315e-01,  ..., -1.0981e-01,\n",
       "           -4.3747e-03, -5.1724e-02],\n",
       "          [ 2.0883e-01, -4.0449e-01,  3.4848e-02,  ...,  1.8332e-01,\n",
       "            4.1884e-01,  1.2690e-01],\n",
       "          [ 2.2043e-01, -4.3417e-01,  1.4145e-01,  ...,  2.5914e-02,\n",
       "            4.4730e-03,  1.6600e-01],\n",
       "          ...,\n",
       "          [-1.7008e-01, -3.3203e-01,  3.2161e-01,  ..., -9.6166e-02,\n",
       "            3.5158e-01,  3.2639e-01],\n",
       "          [ 4.2160e-01, -4.2612e-01, -5.3019e-02,  ...,  1.3938e-01,\n",
       "           -1.2820e-01, -2.5357e-01],\n",
       "          [ 1.0546e-01, -3.6440e-02, -8.6815e-02,  ...,  5.4516e-01,\n",
       "            1.2159e-01, -4.1756e-02]],\n",
       "\n",
       "         [[-4.4678e-01,  2.5297e-01,  4.0313e-01,  ..., -4.2038e-01,\n",
       "            5.3838e-01, -9.8061e-01],\n",
       "          [-4.9524e-01,  6.5838e-01,  6.8314e-01,  ...,  2.0144e-01,\n",
       "            1.7071e-01, -6.8930e-01],\n",
       "          [-4.8590e-01,  2.4467e-01,  2.4403e-01,  ...,  4.2110e-01,\n",
       "            1.3638e-01, -4.0026e-01],\n",
       "          ...,\n",
       "          [-3.4270e-01,  3.8285e-01,  1.0895e-01,  ..., -6.1351e-02,\n",
       "            2.9263e-01, -2.9485e-01],\n",
       "          [-3.4161e-01,  3.6681e-01, -1.4435e-03,  ..., -2.7795e-01,\n",
       "            2.2604e-01, -6.1106e-01],\n",
       "          [-2.9749e-01,  1.8905e-01,  3.2711e-01,  ...,  2.5963e-02,\n",
       "           -4.6245e-02, -8.5759e-01]],\n",
       "\n",
       "         [[ 1.6763e-01, -2.9820e-01,  5.9442e-01,  ..., -8.5589e-02,\n",
       "            1.5804e-01, -1.2082e-01],\n",
       "          [-1.0755e-01, -1.0389e-02,  7.1106e-01,  ...,  5.9947e-02,\n",
       "            4.0488e-01, -1.9036e-01],\n",
       "          [-1.0963e-01,  9.5104e-02,  5.9718e-01,  ...,  3.5435e-02,\n",
       "            3.1637e-01, -8.1538e-02],\n",
       "          ...,\n",
       "          [-1.7077e-01, -1.0975e-01,  7.2461e-01,  ..., -1.0845e-01,\n",
       "            2.1335e-01,  2.8464e-01],\n",
       "          [ 1.6576e-01, -2.3227e-02,  8.1619e-01,  ...,  2.9128e-01,\n",
       "            2.1113e-01, -7.3713e-03],\n",
       "          [ 2.1155e-01, -9.0728e-02,  4.3747e-01,  ..., -4.2082e-02,\n",
       "           -3.4105e-02, -1.2093e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.0140e-01, -2.9497e-02,  1.0492e-01,  ...,  2.5927e-01,\n",
       "           -1.2959e-01,  8.1269e-01],\n",
       "          [-3.8262e-01, -2.4969e-01,  1.9726e-01,  ...,  3.7242e-01,\n",
       "            1.9608e-01,  8.9544e-01],\n",
       "          [-2.3085e-01, -9.1763e-03,  2.9601e-02,  ...,  5.5478e-01,\n",
       "            2.4153e-01,  1.0295e+00],\n",
       "          ...,\n",
       "          [ 5.0778e-02,  9.3049e-02,  4.4827e-02,  ...,  5.2927e-01,\n",
       "            8.3827e-04,  1.1135e+00],\n",
       "          [ 2.0186e-02, -2.3179e-01,  1.2284e-01,  ...,  4.7073e-01,\n",
       "            8.8950e-02,  2.8780e-01],\n",
       "          [ 7.6423e-02, -2.0731e-01,  3.8703e-01,  ...,  6.1068e-01,\n",
       "           -3.3837e-01,  8.2324e-01]],\n",
       "\n",
       "         [[ 6.8481e-03,  2.1314e-01, -1.0354e-01,  ..., -8.2937e-01,\n",
       "           -3.7241e-01,  4.5005e-01],\n",
       "          [ 4.3327e-02,  9.7713e-02, -4.6876e-01,  ..., -7.9236e-01,\n",
       "           -3.3732e-01,  1.0411e-01],\n",
       "          [ 3.9038e-01,  3.7324e-01, -1.2355e-01,  ..., -9.6094e-01,\n",
       "           -4.3769e-01,  1.0221e-01],\n",
       "          ...,\n",
       "          [ 2.9553e-01, -1.6289e-01, -6.6298e-02,  ..., -8.5079e-01,\n",
       "           -3.6460e-01, -1.8373e-01],\n",
       "          [-1.2596e-01,  1.9716e-01,  4.7364e-03,  ..., -8.3690e-01,\n",
       "           -5.2200e-01,  3.0784e-01],\n",
       "          [-4.9398e-02,  5.0919e-01, -4.2195e-01,  ..., -5.9571e-01,\n",
       "           -2.9255e-01,  2.1504e-01]],\n",
       "\n",
       "         [[ 2.0052e-02, -4.6141e-01, -1.2189e-01,  ...,  6.2421e-02,\n",
       "           -2.3487e-01, -2.5918e-01],\n",
       "          [-2.0344e-01,  1.4299e-01,  1.1482e-01,  ...,  6.3621e-01,\n",
       "           -2.8948e-02,  3.8922e-01],\n",
       "          [ 7.8375e-02, -1.6594e-01,  5.0061e-02,  ...,  3.7269e-01,\n",
       "           -4.2626e-01,  2.7204e-01],\n",
       "          ...,\n",
       "          [ 2.4508e-03, -1.3032e-02,  1.8722e-01,  ...,  6.7982e-01,\n",
       "           -4.6660e-01,  5.3120e-01],\n",
       "          [ 2.9350e-02,  1.2167e-01, -3.7634e-01,  ...,  3.5194e-01,\n",
       "           -4.0763e-01,  1.2975e-01],\n",
       "          [ 2.1951e-01, -1.2004e-01, -1.2318e-01,  ...,  3.1889e-01,\n",
       "           -1.3994e-01,  4.6520e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.7021e-01, -1.8252e-01, -3.8274e-01,  ..., -9.8752e-02,\n",
       "           -3.7450e-02, -6.6994e-02],\n",
       "          [ 3.3610e-01, -5.5941e-01,  8.7278e-02,  ..., -2.7412e-01,\n",
       "           -1.4817e-01,  6.7370e-02],\n",
       "          [ 3.9786e-01, -6.2343e-01, -1.2236e-01,  ..., -5.0714e-02,\n",
       "           -2.7413e-01,  1.4116e-01],\n",
       "          ...,\n",
       "          [ 4.3238e-01, -2.1603e-01,  1.7782e-01,  ...,  2.3774e-02,\n",
       "            3.5133e-02,  7.5107e-03],\n",
       "          [ 7.9192e-02, -3.3495e-01, -9.2550e-02,  ...,  2.7567e-01,\n",
       "           -1.2692e-01,  2.9119e-01],\n",
       "          [ 5.2656e-02,  3.4353e-01,  1.0508e-01,  ...,  6.2391e-01,\n",
       "            1.1184e-01, -1.3829e-01]],\n",
       "\n",
       "         [[-4.4419e-01,  2.6252e-01,  3.9330e-01,  ..., -4.1914e-01,\n",
       "            5.4462e-01, -9.6873e-01],\n",
       "          [-6.4836e-01,  4.4696e-01,  3.9195e-01,  ..., -2.7404e-01,\n",
       "            4.0085e-01, -7.4484e-01],\n",
       "          [-5.2451e-01,  4.2251e-01,  4.6249e-01,  ..., -1.4772e-01,\n",
       "            5.3879e-01, -7.3961e-01],\n",
       "          ...,\n",
       "          [-3.9375e-01,  2.4452e-01,  6.1145e-01,  ...,  6.7986e-02,\n",
       "           -5.9113e-02, -4.4663e-01],\n",
       "          [-2.7616e-01,  3.3924e-01,  2.8847e-01,  ..., -1.5412e-01,\n",
       "            3.1020e-01, -9.1350e-01],\n",
       "          [ 1.3094e-01, -3.5754e-02,  4.3908e-01,  ..., -1.6683e-01,\n",
       "           -1.2807e-01, -8.5920e-01]],\n",
       "\n",
       "         [[ 1.6176e-01, -2.7929e-01,  5.9815e-01,  ..., -7.9686e-02,\n",
       "            1.5783e-01, -1.0961e-01],\n",
       "          [-2.1366e-01, -2.9253e-01,  8.0282e-01,  ...,  1.0744e-01,\n",
       "            3.7108e-01, -7.7322e-02],\n",
       "          [ 6.5951e-02, -1.8173e-01,  5.6217e-01,  ...,  1.2356e-01,\n",
       "            3.0846e-01, -1.3002e-01],\n",
       "          ...,\n",
       "          [ 7.2808e-02,  5.6861e-02,  6.9446e-01,  ...,  7.7612e-02,\n",
       "            2.5500e-01, -7.4498e-03],\n",
       "          [ 1.0196e-01, -4.8888e-02,  5.5644e-01,  ...,  1.0205e-01,\n",
       "            9.4534e-02, -4.6549e-02],\n",
       "          [ 3.7252e-01, -1.3026e-01,  4.8814e-01,  ..., -1.1202e-01,\n",
       "           -2.1454e-02, -3.2009e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.9662e-01, -5.8047e-02,  9.1720e-02,  ...,  2.6822e-01,\n",
       "           -1.4468e-01,  7.9447e-01],\n",
       "          [-2.4169e-02, -2.0582e-01, -5.4273e-02,  ...,  5.3294e-01,\n",
       "            1.5324e-01,  2.6322e-01],\n",
       "          [-9.1494e-02, -2.0693e-01,  8.2308e-02,  ...,  4.7028e-01,\n",
       "           -1.8725e-02,  2.8974e-01],\n",
       "          ...,\n",
       "          [ 6.6749e-02, -5.2992e-01,  9.8081e-03,  ...,  4.8628e-01,\n",
       "            1.8268e-01,  3.5376e-01],\n",
       "          [ 6.7461e-02, -1.4060e-01,  1.2198e-01,  ...,  3.3892e-01,\n",
       "           -3.0574e-01,  4.8238e-01],\n",
       "          [-4.1448e-01, -4.5692e-01,  4.4610e-01,  ...,  3.9524e-01,\n",
       "           -4.7749e-01,  6.1677e-01]],\n",
       "\n",
       "         [[ 7.8998e-03,  2.3534e-01, -8.8958e-02,  ..., -8.6303e-01,\n",
       "           -3.6366e-01,  4.3910e-01],\n",
       "          [ 1.6220e-01,  3.3686e-01, -1.0325e-01,  ..., -8.1698e-01,\n",
       "           -5.0204e-01,  2.2557e-01],\n",
       "          [ 1.0170e-01,  3.3991e-01, -2.3578e-01,  ..., -9.1858e-01,\n",
       "           -6.6989e-01,  3.2135e-01],\n",
       "          ...,\n",
       "          [ 3.9440e-01,  1.4360e-01, -3.3363e-02,  ..., -6.9237e-01,\n",
       "           -6.0124e-01,  2.3707e-01],\n",
       "          [-8.5726e-02,  5.1106e-01, -4.1606e-01,  ..., -9.0287e-01,\n",
       "           -6.0134e-01,  1.7853e-01],\n",
       "          [ 4.9855e-01,  6.3601e-01, -3.9610e-01,  ..., -3.4947e-01,\n",
       "            3.9610e-02, -2.6438e-02]],\n",
       "\n",
       "         [[ 1.1212e-02, -4.7221e-01, -1.1307e-01,  ...,  5.0494e-02,\n",
       "           -2.1895e-01, -2.5762e-01],\n",
       "          [ 1.0567e-01, -2.6088e-01, -1.1581e-01,  ...,  2.6089e-01,\n",
       "           -5.4238e-01,  3.7008e-02],\n",
       "          [ 1.2299e-01, -1.9742e-01, -7.7565e-02,  ...,  1.3593e-01,\n",
       "           -4.1375e-01,  1.2646e-01],\n",
       "          ...,\n",
       "          [ 7.3904e-02,  1.5715e-01, -7.3852e-02,  ..., -6.7121e-02,\n",
       "           -5.1068e-01, -7.7314e-03],\n",
       "          [ 8.9849e-02, -1.3781e-01, -2.5142e-01,  ...,  3.0486e-01,\n",
       "           -1.0401e-01,  2.9477e-01],\n",
       "          [ 1.7253e-01, -1.5255e-01, -9.6090e-02,  ...,  1.7914e-01,\n",
       "           -2.5207e-01,  1.8061e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 3.5268e-01,  1.4549e-01, -9.4480e-01,  ..., -3.4653e-01,\n",
       "            1.0034e+00,  1.9760e-02],\n",
       "          [ 2.6534e+00,  2.2978e+00, -5.7474e-01,  ..., -1.4471e-01,\n",
       "           -1.1016e+00, -1.0150e+00],\n",
       "          [ 1.2266e+00,  3.1737e+00, -6.5705e-01,  ..., -1.3604e+00,\n",
       "           -5.1805e+00,  9.2022e-01],\n",
       "          ...,\n",
       "          [ 1.0425e+00,  2.1066e+00,  2.3837e-01,  ..., -1.5675e+00,\n",
       "            1.1573e+00, -9.2887e-01],\n",
       "          [ 1.8409e+00,  3.9317e+00, -1.4425e-01,  ..., -2.6941e+00,\n",
       "           -4.4625e+00, -1.1118e+00],\n",
       "          [-2.5508e+00, -4.0474e+00,  1.6715e+00,  ...,  7.1841e-01,\n",
       "            2.5843e+00,  3.1389e+00]],\n",
       "\n",
       "         [[ 7.6584e-01,  1.2353e+00,  2.3772e+00,  ...,  9.3637e-01,\n",
       "           -1.2296e+00,  5.4176e-02],\n",
       "          [-2.6165e-01, -2.3007e+00,  1.9700e+00,  ...,  1.8377e+00,\n",
       "           -3.6526e+00, -4.3984e-01],\n",
       "          [ 1.1024e+00,  5.8979e-01,  1.1989e+00,  ...,  1.2787e+00,\n",
       "           -2.4829e+00, -1.0348e+00],\n",
       "          ...,\n",
       "          [-7.6352e-01,  6.4071e-01,  1.2960e+00,  ...,  1.4256e+00,\n",
       "           -5.0761e+00, -1.2026e-01],\n",
       "          [ 1.0319e+00, -9.5882e-01,  6.4149e-01,  ..., -3.2688e-01,\n",
       "           -2.4427e+00,  1.1057e+00],\n",
       "          [ 1.4285e+00,  4.1816e+00,  2.6945e-01,  ..., -2.8521e+00,\n",
       "            3.6544e-01,  2.5597e+00]],\n",
       "\n",
       "         [[ 2.8309e+00,  4.0855e-01,  2.1707e+00,  ..., -3.2816e+00,\n",
       "           -3.6693e-01,  5.3578e+00],\n",
       "          [ 9.7459e-02,  1.0303e+00, -1.5366e+00,  ..., -3.3582e+00,\n",
       "            2.5581e-01,  1.5945e+00],\n",
       "          [ 1.3309e+00,  6.0277e-01,  1.5801e-01,  ..., -2.5418e+00,\n",
       "           -4.2647e-02,  2.6879e+00],\n",
       "          ...,\n",
       "          [ 1.7628e-01,  5.4776e-01,  1.3873e+00,  ..., -2.7516e+00,\n",
       "            1.2895e+00,  1.6892e+00],\n",
       "          [-1.2517e+00, -1.6373e-01,  9.0250e-01,  ..., -3.9269e-01,\n",
       "            2.1327e+00,  1.4162e+00],\n",
       "          [-6.7468e-01, -1.0667e+00,  1.0497e+00,  ...,  6.3148e-01,\n",
       "           -3.0002e-01, -2.7657e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.9584e-01, -3.9867e+00, -1.4539e-01,  ...,  3.8592e-01,\n",
       "           -3.6787e+00,  7.9120e-01],\n",
       "          [-1.1718e+00, -1.1634e+00, -8.9186e-01,  ...,  3.3724e+00,\n",
       "            1.2422e+00,  2.2802e+00],\n",
       "          [ 4.7395e-01, -7.5251e-01,  1.3880e-02,  ...,  4.2306e+00,\n",
       "            2.5253e+00,  1.2742e+00],\n",
       "          ...,\n",
       "          [ 2.0818e-01,  9.0395e-04, -5.6924e-01,  ...,  4.2195e+00,\n",
       "            3.4725e+00,  1.9295e+00],\n",
       "          [ 1.0369e+00,  1.1357e+00, -1.2364e-01,  ...,  2.7723e+00,\n",
       "            2.7364e+00,  1.0512e+00],\n",
       "          [ 2.2933e-01,  1.7577e-01, -8.3879e-01,  ..., -1.5954e+00,\n",
       "           -3.4988e-01,  1.0834e+00]],\n",
       "\n",
       "         [[-2.5018e+00, -5.6768e-01,  1.2152e+00,  ..., -2.8697e-01,\n",
       "            1.2575e+00,  2.1790e+00],\n",
       "          [ 2.7126e-01,  8.0218e-01,  1.3027e+00,  ...,  3.2112e-02,\n",
       "           -2.4678e-01,  1.4147e+00],\n",
       "          [ 1.5709e+00,  2.6861e+00,  1.8381e+00,  ...,  1.0035e+00,\n",
       "            1.8356e-01, -2.1165e-01],\n",
       "          ...,\n",
       "          [ 2.0351e+00,  1.9214e+00,  1.8484e+00,  ...,  1.5918e+00,\n",
       "            5.4550e-02,  3.4848e-01],\n",
       "          [ 1.6104e+00,  1.3247e+00, -4.6673e-01,  ...,  8.8815e-01,\n",
       "            5.6864e-01,  1.7490e+00],\n",
       "          [-5.2554e+00, -3.5182e+00, -8.8171e-02,  ..., -2.6845e+00,\n",
       "            2.1069e+00,  1.1004e+00]],\n",
       "\n",
       "         [[-1.1712e+00, -2.4198e+00, -7.7503e-01,  ...,  3.0502e-01,\n",
       "           -3.4094e-01, -2.9410e-01],\n",
       "          [ 2.0455e+00, -3.7093e+00, -2.7987e+00,  ..., -2.7894e-01,\n",
       "           -9.9548e-01, -3.2345e+00],\n",
       "          [-8.4640e-01, -1.2621e+00,  6.7639e-01,  ...,  8.2996e-01,\n",
       "           -1.2016e+00, -4.2103e-01],\n",
       "          ...,\n",
       "          [ 1.1070e+00, -1.1163e+00,  1.1431e+00,  ...,  4.3161e-02,\n",
       "           -2.2043e-03,  3.4828e-01],\n",
       "          [-3.4936e-02, -7.0557e-01, -1.7033e-01,  ...,  1.1285e+00,\n",
       "            1.1856e+00,  1.5193e+00],\n",
       "          [-1.9674e+00,  2.5957e-02,  4.0078e-01,  ..., -9.0635e-01,\n",
       "           -1.8829e+00,  2.7280e+00]]],\n",
       "\n",
       "\n",
       "        [[[-7.2749e-02,  1.3081e+00,  1.9806e+00,  ..., -2.1510e+00,\n",
       "            6.3306e-01,  2.0775e+00],\n",
       "          [-4.6394e-01,  4.4163e+00, -1.7372e-01,  ..., -1.5378e+00,\n",
       "           -3.4707e+00,  7.6803e-02],\n",
       "          [ 9.9889e-01,  4.6803e+00,  1.4102e+00,  ..., -1.0205e+00,\n",
       "           -2.9372e+00, -2.6821e+00],\n",
       "          ...,\n",
       "          [ 3.2125e+00,  1.1862e+00, -2.9585e-01,  ...,  3.2796e-01,\n",
       "            7.6725e-01, -5.9877e-01],\n",
       "          [ 2.3039e+00,  7.3197e-01, -2.4939e-02,  ..., -7.4157e-01,\n",
       "            6.1911e-01,  1.3080e+00],\n",
       "          [-2.5214e+00, -3.9887e+00,  1.6642e+00,  ...,  7.6987e-01,\n",
       "            2.7238e+00,  3.2448e+00]],\n",
       "\n",
       "         [[-3.2622e+00,  1.0168e+00,  1.5396e+00,  ...,  3.1018e+00,\n",
       "           -3.2378e+00, -3.3561e+00],\n",
       "          [ 1.7002e+00, -9.1367e-02,  2.3228e-01,  ...,  1.7971e+00,\n",
       "           -4.1718e+00, -2.6633e+00],\n",
       "          [ 1.6745e+00,  1.3823e+00,  3.6333e-01,  ...,  1.9851e+00,\n",
       "           -1.4379e+00, -1.2473e+00],\n",
       "          ...,\n",
       "          [-1.8846e-02,  1.5700e+00, -1.1913e+00,  ...,  1.1655e+00,\n",
       "            5.2751e-01,  7.0789e-01],\n",
       "          [-1.8493e+00,  1.3141e+00, -7.6135e-01,  ..., -8.3344e-02,\n",
       "           -1.3978e+00,  1.6629e+00],\n",
       "          [ 1.2999e+00,  4.2752e+00,  2.0184e-01,  ..., -2.8793e+00,\n",
       "            3.4483e-01,  2.5386e+00]],\n",
       "\n",
       "         [[-1.6679e+00, -2.2154e+00,  3.1543e+00,  ..., -2.2250e+00,\n",
       "            1.4436e+00,  2.0979e+00],\n",
       "          [-4.3505e-01, -4.5091e-01,  2.4711e+00,  ..., -1.6022e+00,\n",
       "            1.6996e-01,  3.2067e+00],\n",
       "          [-2.2590e+00, -1.8799e+00,  9.7127e-01,  ..., -4.6062e-01,\n",
       "            2.0945e+00,  1.6991e+00],\n",
       "          ...,\n",
       "          [-1.2654e+00,  1.2572e+00,  1.1365e+00,  ..., -5.6109e-04,\n",
       "            1.8440e+00,  1.8696e+00],\n",
       "          [-1.0082e+00,  2.2518e+00, -4.2995e-01,  ...,  6.3005e-01,\n",
       "           -1.7266e-01,  7.9741e-01],\n",
       "          [-6.7043e-01, -1.0381e+00,  1.0014e+00,  ...,  6.7441e-01,\n",
       "           -3.5368e-01, -2.4393e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.8727e+00, -1.0014e+00, -4.0059e+00,  ...,  3.5362e+00,\n",
       "           -1.4454e+00,  8.9440e-01],\n",
       "          [ 3.8149e-01,  5.8231e-01, -1.2199e+00,  ...,  5.1012e+00,\n",
       "            3.0008e+00,  9.2370e-01],\n",
       "          [ 8.1250e-01,  5.1847e-01,  2.3626e+00,  ...,  4.8234e+00,\n",
       "           -1.0791e+00,  1.5325e+00],\n",
       "          ...,\n",
       "          [ 5.2367e-01,  2.1328e-01,  1.9158e-01,  ...,  3.2801e+00,\n",
       "            1.8686e+00, -8.6850e-01],\n",
       "          [-6.0714e-01,  1.5321e+00, -2.0348e+00,  ...,  1.7543e+00,\n",
       "            2.0031e+00,  7.7432e-01],\n",
       "          [ 1.6892e-01,  1.0760e-01, -8.2352e-01,  ..., -1.5720e+00,\n",
       "           -3.5142e-01,  1.1183e+00]],\n",
       "\n",
       "         [[-1.2939e+00,  8.8748e-01,  6.1499e-01,  ...,  1.1873e+00,\n",
       "           -2.1574e+00,  1.4652e-01],\n",
       "          [ 1.9396e+00,  2.5285e+00,  1.9966e+00,  ...,  2.1647e+00,\n",
       "            1.7644e-01,  8.2747e-01],\n",
       "          [ 2.5735e+00,  1.9818e+00,  5.0138e-01,  ...,  1.3437e+00,\n",
       "           -5.7515e-01,  3.6480e+00],\n",
       "          ...,\n",
       "          [-2.8652e+00,  1.7158e+00,  3.0879e+00,  ...,  2.5333e-01,\n",
       "           -4.1689e-01,  1.6493e+00],\n",
       "          [-1.1045e+00, -7.0984e-01, -4.2938e-02,  ...,  1.0822e+00,\n",
       "            2.9945e-01,  1.7291e+00],\n",
       "          [-5.3987e+00, -3.5387e+00, -7.9600e-03,  ..., -2.6696e+00,\n",
       "            2.1095e+00,  1.0780e+00]],\n",
       "\n",
       "         [[ 1.9870e+00,  2.2185e-01,  2.5206e+00,  ...,  2.5709e+00,\n",
       "           -1.6983e+00,  1.4534e+00],\n",
       "          [-1.0072e+00, -2.0028e+00, -3.0949e-01,  ...,  7.3115e-01,\n",
       "           -1.1170e+00,  1.6755e+00],\n",
       "          [ 9.6276e-01, -1.6388e+00, -3.8276e+00,  ...,  1.9129e+00,\n",
       "            3.6280e-01,  4.9359e-01],\n",
       "          ...,\n",
       "          [ 1.7607e+00,  1.9272e+00,  3.2390e-01,  ...,  1.7859e+00,\n",
       "           -1.7847e+00, -1.5978e+00],\n",
       "          [ 1.3830e+00, -1.1008e-02,  6.0434e-01,  ...,  3.9443e-01,\n",
       "           -2.4158e+00, -3.7381e-01],\n",
       "          [-1.9496e+00,  7.6700e-02,  3.7605e-01,  ..., -9.3222e-01,\n",
       "           -1.9603e+00,  2.6410e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 1.2206e-01,  1.0080e+00,  1.8610e+00,  ...,  1.8707e+00,\n",
       "            6.1597e-02,  7.0601e-01],\n",
       "          [-4.4225e-01,  1.0273e+00,  1.6201e+00,  ..., -3.4343e-01,\n",
       "           -3.7998e-01,  1.2009e+00],\n",
       "          [-2.8002e+00, -3.5199e-01,  4.3544e+00,  ..., -1.5592e+00,\n",
       "            3.0703e+00, -1.9713e+00],\n",
       "          ...,\n",
       "          [-2.0928e+00,  3.5622e-01,  1.7550e+00,  ..., -5.5781e-01,\n",
       "           -1.5819e-01,  2.7588e+00],\n",
       "          [-1.4466e+00,  2.2735e+00,  4.2108e-01,  ..., -2.2593e+00,\n",
       "            4.8225e-01, -4.9104e-01],\n",
       "          [-2.5948e+00, -4.0139e+00,  1.7422e+00,  ...,  6.9015e-01,\n",
       "            2.7047e+00,  3.2358e+00]],\n",
       "\n",
       "         [[-2.3719e+00,  1.9762e+00, -2.0494e-01,  ...,  1.6767e+00,\n",
       "           -4.8689e+00, -6.3281e-01],\n",
       "          [-3.4812e-02,  3.8130e+00,  6.3929e-01,  ...,  2.1667e+00,\n",
       "           -2.5056e+00, -1.1911e-01],\n",
       "          [-1.9656e-02,  1.6836e+00, -1.9081e+00,  ..., -9.7811e-01,\n",
       "           -5.4147e-01, -1.1065e+00],\n",
       "          ...,\n",
       "          [-2.5046e+00,  2.1009e+00,  1.4956e+00,  ..., -1.1109e+00,\n",
       "           -1.9895e+00,  2.1873e+00],\n",
       "          [-6.7078e-01, -1.2101e+00, -7.5453e-01,  ..., -2.3079e-01,\n",
       "           -1.7073e+00,  9.3739e-01],\n",
       "          [ 1.3467e+00,  4.2115e+00,  1.8604e-01,  ..., -2.8516e+00,\n",
       "            4.2716e-01,  2.5355e+00]],\n",
       "\n",
       "         [[-2.2551e+00,  5.7937e-01, -5.6986e-01,  ..., -2.1177e+00,\n",
       "           -6.6260e-01,  1.0733e+00],\n",
       "          [-1.4298e+00,  2.9073e+00, -2.7928e+00,  ...,  7.9589e-01,\n",
       "           -1.4057e+00,  1.2756e+00],\n",
       "          [-5.8473e-01, -3.8232e-01, -1.3471e+00,  ..., -4.0213e+00,\n",
       "           -2.0873e+00, -1.3331e+00],\n",
       "          ...,\n",
       "          [-8.7877e-02,  2.0009e+00,  1.5228e+00,  ..., -1.2256e+00,\n",
       "            6.9932e-01,  3.2851e+00],\n",
       "          [-2.0666e+00,  1.2304e+00, -1.8928e-02,  ..., -8.1951e-01,\n",
       "           -1.0339e+00,  1.6461e+00],\n",
       "          [-6.8481e-01, -1.0060e+00,  1.0047e+00,  ...,  6.7497e-01,\n",
       "           -3.8739e-01, -2.1196e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-5.7995e-01,  1.8143e+00, -7.1740e-01,  ...,  6.4983e-01,\n",
       "            1.3015e+00, -1.6111e-01],\n",
       "          [ 1.2528e+00,  2.0634e+00,  7.1334e-01,  ..., -1.5964e-01,\n",
       "           -1.5859e+00,  1.7819e+00],\n",
       "          [-1.2137e+00, -1.7432e+00, -9.1661e-01,  ..., -1.2421e+00,\n",
       "           -1.0519e+00, -5.5554e-01],\n",
       "          ...,\n",
       "          [ 1.1090e-02,  3.3525e-01, -1.7439e-01,  ..., -7.3163e-01,\n",
       "           -1.5662e+00,  3.2171e+00],\n",
       "          [-4.8952e-01,  1.7002e+00, -9.5518e-01,  ...,  1.2873e+00,\n",
       "            2.5466e-01, -3.5898e-01],\n",
       "          [ 1.6583e-01,  1.7044e-01, -8.0539e-01,  ..., -1.6471e+00,\n",
       "           -3.4105e-01,  1.0566e+00]],\n",
       "\n",
       "         [[ 8.6426e-01,  1.0078e+00,  2.5827e+00,  ...,  1.5415e+00,\n",
       "           -2.8132e-01,  7.6934e-01],\n",
       "          [ 7.1484e-01,  7.5000e-01, -7.2871e-01,  ..., -4.0599e-02,\n",
       "           -2.2984e-01,  2.9347e-01],\n",
       "          [-5.9382e-01,  1.3352e+00,  6.0311e-01,  ..., -4.1417e-01,\n",
       "           -2.8967e+00, -1.4351e+00],\n",
       "          ...,\n",
       "          [-1.3740e+00,  6.1979e-01, -4.8524e-01,  ..., -2.4678e-01,\n",
       "           -6.6074e-01, -4.4933e-01],\n",
       "          [-2.1696e+00,  1.7001e+00, -4.5930e-01,  ...,  3.1794e-01,\n",
       "            6.3555e-01,  1.4808e+00],\n",
       "          [-5.3658e+00, -3.5198e+00, -9.2473e-02,  ..., -2.6947e+00,\n",
       "            2.1430e+00,  1.1387e+00]],\n",
       "\n",
       "         [[ 1.4551e+00, -1.4637e+00, -4.0087e+00,  ...,  1.7150e+00,\n",
       "           -4.6699e-01, -1.4135e+00],\n",
       "          [-6.7061e-01, -1.3664e+00, -6.0925e-01,  ..., -1.3969e+00,\n",
       "           -2.7737e+00, -2.2353e-01],\n",
       "          [ 2.7080e+00,  5.8017e-02, -1.5588e+00,  ...,  1.4575e+00,\n",
       "           -1.1388e+00,  4.0936e+00],\n",
       "          ...,\n",
       "          [-1.3471e+00, -3.7904e+00,  1.3042e+00,  ...,  1.3528e+00,\n",
       "            1.0290e+00, -3.6133e-02],\n",
       "          [ 5.9795e-01, -2.5172e+00,  9.2401e-01,  ...,  1.1457e+00,\n",
       "            1.5312e+00,  2.0518e+00],\n",
       "          [-1.9665e+00,  2.9180e-02,  3.8401e-01,  ..., -8.8281e-01,\n",
       "           -1.9889e+00,  2.6692e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[-2.9103e+00,  1.2548e+00, -1.7461e+00,  ..., -2.5542e-01,\n",
       "           -4.0950e-01, -5.8219e-01],\n",
       "          [-1.1348e+00,  3.9054e-01,  1.2058e+00,  ..., -6.4495e-01,\n",
       "           -1.6144e+00, -4.3617e-01],\n",
       "          [-7.8951e-01, -3.3885e-01,  5.4855e-02,  ..., -4.1497e-01,\n",
       "           -1.9351e+00, -2.7169e-01],\n",
       "          ...,\n",
       "          [-9.3695e-02,  9.4197e-01, -4.4290e-01,  ..., -1.2741e+00,\n",
       "           -1.1444e+00,  9.9145e-01],\n",
       "          [ 3.0190e-01, -5.5309e-01, -6.7363e-01,  ..., -1.1843e+00,\n",
       "           -8.2802e-01, -6.4259e-01],\n",
       "          [ 2.5886e-01,  5.0432e-01, -3.5307e-01,  ..., -2.1546e-02,\n",
       "           -3.3159e-01, -4.7964e-01]],\n",
       "\n",
       "         [[-3.2382e-01, -3.1409e-01, -4.6576e-01,  ..., -2.7452e+00,\n",
       "           -3.9896e-01,  1.5074e+00],\n",
       "          [ 1.1938e-01, -3.2352e-01,  1.0166e+00,  ..., -5.7785e-01,\n",
       "           -4.2509e-01, -1.6352e+00],\n",
       "          [ 2.8422e-01, -5.4650e-01,  1.7940e+00,  ...,  7.3378e-01,\n",
       "           -8.2195e-01, -3.8083e-01],\n",
       "          ...,\n",
       "          [-6.7013e-01,  1.1190e+00,  1.4996e+00,  ..., -3.9955e-02,\n",
       "           -1.5127e+00, -2.3912e+00],\n",
       "          [-1.2935e+00,  6.7207e-01,  8.9680e-01,  ...,  1.3603e+00,\n",
       "           -6.3766e-01, -7.5409e-01],\n",
       "          [ 9.8369e-02, -2.9616e-01,  2.8589e-01,  ...,  1.3778e-01,\n",
       "           -5.5305e-01, -1.7893e-01]],\n",
       "\n",
       "         [[ 6.4258e-01,  3.4168e+00, -1.8090e+00,  ...,  1.5648e+00,\n",
       "            3.0541e+00,  1.2192e-01],\n",
       "          [ 3.2088e-01,  9.3068e-01, -8.9481e-01,  ...,  1.8167e+00,\n",
       "           -6.3150e-01, -8.7434e-01],\n",
       "          [ 4.9024e-01,  1.5783e+00,  6.3168e-01,  ...,  2.7310e+00,\n",
       "           -4.3824e-01, -1.0908e+00],\n",
       "          ...,\n",
       "          [-4.6199e-01,  1.0427e+00, -1.8830e+00,  ...,  1.6372e+00,\n",
       "           -1.0159e+00,  5.2258e-02],\n",
       "          [ 2.9828e-01,  1.6060e+00,  1.0836e+00,  ..., -9.7432e-01,\n",
       "           -1.1934e-01,  3.2550e-01],\n",
       "          [ 6.3414e-01, -1.5759e-02, -2.7612e-01,  ...,  2.7897e-01,\n",
       "           -9.1669e-03, -7.5047e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-7.1431e-01, -1.0013e+00, -6.0058e-01,  ..., -7.5556e-01,\n",
       "            1.3236e+00, -1.5751e+00],\n",
       "          [ 4.0667e-01, -3.3271e-01,  9.0510e-01,  ..., -3.4917e-01,\n",
       "           -5.4363e-01, -1.7788e+00],\n",
       "          [ 1.7244e+00,  1.5847e-01, -2.8119e-02,  ...,  5.7428e-01,\n",
       "            6.7815e-01,  1.6799e-01],\n",
       "          ...,\n",
       "          [ 8.1298e-01, -2.5322e-01, -2.3984e-01,  ...,  2.0258e+00,\n",
       "           -8.6830e-01, -6.9661e-01],\n",
       "          [-4.5501e-01,  1.0177e+00, -5.3470e-01,  ..., -1.2103e-01,\n",
       "            4.8764e-01, -4.5130e-01],\n",
       "          [-6.9890e-01,  3.7037e-01, -4.0106e-02,  ...,  6.3795e-01,\n",
       "            8.9346e-01, -4.3924e-01]],\n",
       "\n",
       "         [[-7.3737e-01, -8.6645e-02, -1.1850e-01,  ..., -3.5103e-01,\n",
       "            5.7729e-01,  2.1713e+00],\n",
       "          [ 3.1621e-01,  1.1863e+00, -1.2193e+00,  ..., -1.6378e+00,\n",
       "           -2.7618e-01,  3.8738e-01],\n",
       "          [ 9.4206e-01,  8.0761e-01, -6.7200e-01,  ..., -9.7103e-01,\n",
       "           -1.4820e-01,  1.8393e+00],\n",
       "          ...,\n",
       "          [ 3.9993e-01, -3.1442e-01, -1.8543e-01,  ..., -2.0601e-01,\n",
       "            3.9271e-01, -1.1639e+00],\n",
       "          [ 2.5948e-01, -2.5792e-01, -7.3139e-01,  ..., -2.6646e-01,\n",
       "            8.0958e-01, -2.2009e-01],\n",
       "          [ 5.9099e-02,  2.6191e-01, -1.1293e-02,  ..., -7.9603e-01,\n",
       "            1.6785e-01,  5.9224e-01]],\n",
       "\n",
       "         [[-6.9646e-01, -1.7541e+00,  2.0390e+00,  ..., -9.4757e-01,\n",
       "           -3.3054e+00, -1.2304e+00],\n",
       "          [-2.7969e-01, -1.2601e+00,  1.4865e+00,  ..., -9.6996e-02,\n",
       "           -2.4670e-01, -4.9623e-01],\n",
       "          [-2.8458e-01, -1.4027e-01,  1.2575e+00,  ..., -3.4389e-01,\n",
       "           -3.0611e-01, -2.2350e+00],\n",
       "          ...,\n",
       "          [-1.7666e-01,  9.6668e-01,  1.7725e+00,  ...,  2.5039e-02,\n",
       "            5.6022e-01, -7.9219e-01],\n",
       "          [ 4.5000e-01, -3.8700e-01,  5.0687e-01,  ..., -8.7822e-01,\n",
       "           -4.2556e-01, -1.3116e+00],\n",
       "          [ 5.9638e-01,  1.0141e-01, -1.3383e-01,  ...,  2.6545e-01,\n",
       "           -1.7726e-01,  2.2009e-01]]],\n",
       "\n",
       "\n",
       "        [[[-3.2695e-01,  1.7732e+00, -1.8523e+00,  ..., -1.3590e+00,\n",
       "            1.9212e-01,  6.9370e-01],\n",
       "          [-7.4461e-01,  1.0416e-01, -9.0041e-01,  ..., -2.3682e-01,\n",
       "           -1.4692e+00,  1.2031e+00],\n",
       "          [ 6.3541e-01, -5.8766e-01,  2.9372e-01,  ..., -2.6594e+00,\n",
       "           -1.5730e+00,  1.3519e+00],\n",
       "          ...,\n",
       "          [-7.9119e-01,  4.5402e-01, -2.0449e-01,  ..., -1.4078e+00,\n",
       "           -7.6026e-01,  1.8288e-01],\n",
       "          [ 2.3916e-01, -2.9870e-02, -1.5390e+00,  ..., -2.2228e-02,\n",
       "           -1.1764e+00, -1.5471e+00],\n",
       "          [ 2.5360e-01,  5.1973e-01, -3.4710e-01,  ..., -3.8110e-03,\n",
       "           -3.5753e-01, -5.4800e-01]],\n",
       "\n",
       "         [[ 8.8743e-01, -1.9470e-01,  4.1490e-01,  ...,  1.6215e+00,\n",
       "            9.8154e-01, -1.9180e+00],\n",
       "          [ 4.4308e-01, -7.0361e-01,  2.2240e+00,  ...,  1.3447e+00,\n",
       "           -4.2757e-01, -9.0834e-01],\n",
       "          [-1.7491e-01,  5.8503e-02,  6.8340e-01,  ...,  1.0988e+00,\n",
       "            7.0657e-01, -1.8701e-01],\n",
       "          ...,\n",
       "          [-6.7238e-01,  6.0794e-01, -7.8502e-01,  ..., -1.4142e+00,\n",
       "            1.5565e-01,  2.7009e-01],\n",
       "          [-4.2816e-01,  5.4101e-02,  8.3810e-01,  ..., -1.5272e-01,\n",
       "            6.0425e-01, -1.0115e+00],\n",
       "          [ 1.1890e-01, -3.2188e-01,  2.6370e-01,  ...,  1.3356e-01,\n",
       "           -5.6559e-01, -1.7688e-01]],\n",
       "\n",
       "         [[-5.3030e-01,  4.0404e-01, -3.9665e-01,  ...,  2.1698e+00,\n",
       "            3.6858e-01, -8.5391e-02],\n",
       "          [ 2.6034e-01,  1.3926e+00,  4.2253e-01,  ...,  1.9402e+00,\n",
       "           -8.8448e-01, -1.1645e+00],\n",
       "          [-5.2259e-01, -8.0954e-01,  1.2834e+00,  ...,  9.2503e-01,\n",
       "           -1.4814e+00,  5.6874e-01],\n",
       "          ...,\n",
       "          [-1.0027e+00,  1.8700e+00, -2.1599e-01,  ...,  1.6739e+00,\n",
       "           -1.3785e-02,  3.6723e-01],\n",
       "          [-2.5343e-01,  1.9501e+00,  7.9521e-03,  ...,  5.0605e-01,\n",
       "            3.8182e-02,  9.3275e-01],\n",
       "          [ 6.2948e-01, -2.3945e-02, -2.7067e-01,  ...,  2.8721e-01,\n",
       "           -1.2200e-02, -4.2143e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.5011e+00,  1.6163e+00, -4.7381e-01,  ...,  1.6531e+00,\n",
       "            1.1913e+00,  2.2190e-01],\n",
       "          [ 1.3474e+00, -1.5239e-01,  1.1234e+00,  ...,  2.9672e-01,\n",
       "            3.6053e-01,  3.1525e-01],\n",
       "          [-7.3155e-01,  1.8031e+00,  1.7085e+00,  ...,  1.4496e+00,\n",
       "            3.8317e-02,  6.0182e-01],\n",
       "          ...,\n",
       "          [ 1.1621e+00,  1.7528e-01,  3.7428e-01,  ..., -3.6158e+00,\n",
       "            1.3468e+00, -7.5130e-01],\n",
       "          [ 7.4641e-01,  2.6403e-01, -5.9175e-01,  ..., -1.3115e+00,\n",
       "            3.7555e-01,  2.1421e-01],\n",
       "          [-6.9988e-01,  3.5710e-01, -1.1679e-02,  ...,  6.0515e-01,\n",
       "            8.7835e-01, -4.3379e-01]],\n",
       "\n",
       "         [[-3.0708e-01, -2.2852e-01, -9.8053e-01,  ..., -1.8458e+00,\n",
       "            2.9831e-01,  8.6148e-01],\n",
       "          [ 1.0169e+00,  4.2548e-03, -8.9750e-02,  ..., -2.5816e-01,\n",
       "            4.2961e-01,  1.3183e+00],\n",
       "          [-2.2103e-02, -5.6890e-01,  1.4456e+00,  ...,  1.5811e-01,\n",
       "            2.3867e-02,  1.0475e+00],\n",
       "          ...,\n",
       "          [ 4.4565e-01,  7.4104e-01, -5.8800e-01,  ..., -2.3464e-01,\n",
       "           -4.5057e-01,  1.7121e+00],\n",
       "          [-1.2558e-01,  5.3623e-01, -5.9713e-01,  ..., -8.5063e-01,\n",
       "           -1.0746e+00,  5.6093e-01],\n",
       "          [ 6.2456e-02,  2.7703e-01, -2.5484e-02,  ..., -8.0609e-01,\n",
       "            1.4047e-01,  6.2413e-01]],\n",
       "\n",
       "         [[ 7.3768e-01, -9.9852e-02,  2.0660e+00,  ...,  4.5416e-01,\n",
       "            4.5231e-01, -8.5649e-01],\n",
       "          [-8.5978e-01,  2.8571e-02,  1.2511e+00,  ..., -2.5090e-01,\n",
       "            1.2473e-01, -2.2680e+00],\n",
       "          [-3.5287e-01,  1.8717e-01,  2.1867e-01,  ..., -1.2893e+00,\n",
       "            6.5745e-01, -1.2686e+00],\n",
       "          ...,\n",
       "          [-8.7540e-01, -9.5396e-01, -4.5767e-01,  ..., -1.3757e+00,\n",
       "           -6.1740e-01,  7.1408e-01],\n",
       "          [ 6.2128e-02, -1.1164e+00,  3.3914e-01,  ..., -8.0750e-01,\n",
       "           -1.7405e-01,  1.6060e-01],\n",
       "          [ 6.2432e-01,  7.6529e-02, -1.2156e-01,  ...,  2.9629e-01,\n",
       "           -1.8978e-01,  2.2398e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.4404e+00, -2.0357e-02, -3.7426e-01,  ..., -2.8193e-01,\n",
       "           -2.1191e+00,  1.5385e+00],\n",
       "          [ 4.9110e-01,  1.2197e+00, -8.7785e-01,  ...,  8.0911e-01,\n",
       "           -9.4800e-01,  8.1619e-01],\n",
       "          [ 4.5883e-01,  1.9923e+00, -3.8855e-02,  ..., -1.1617e+00,\n",
       "           -3.1417e-01, -7.6320e-02],\n",
       "          ...,\n",
       "          [ 5.5292e-01,  1.6805e+00, -2.5909e+00,  ..., -1.1640e+00,\n",
       "            1.4451e-01, -1.3570e+00],\n",
       "          [-4.6177e-01,  2.7343e-01, -1.0185e+00,  ...,  2.9091e-01,\n",
       "           -2.0304e-03, -5.2764e-01],\n",
       "          [ 2.4842e-01,  5.4211e-01, -3.3196e-01,  ..., -4.4676e-02,\n",
       "           -3.3128e-01, -5.0425e-01]],\n",
       "\n",
       "         [[ 1.2170e-01,  4.5089e-02,  2.1358e+00,  ...,  1.8243e-01,\n",
       "            2.0102e+00,  1.7343e-01],\n",
       "          [ 4.2178e-01, -5.3570e-01,  3.4781e-01,  ...,  1.3776e+00,\n",
       "            1.0005e+00, -5.3166e-01],\n",
       "          [ 8.8445e-01, -2.3255e-03,  5.1934e-01,  ...,  1.1364e+00,\n",
       "           -1.8915e+00, -7.7510e-01],\n",
       "          ...,\n",
       "          [ 4.8989e-01, -7.1700e-02, -3.6507e-01,  ..., -1.7957e+00,\n",
       "           -3.4016e-01, -3.1730e-01],\n",
       "          [-1.5211e-01, -4.1154e-01, -4.6516e-02,  ...,  5.1829e-01,\n",
       "           -7.1232e-02, -8.6920e-01],\n",
       "          [ 1.3010e-01, -2.8885e-01,  2.5779e-01,  ...,  1.5271e-01,\n",
       "           -5.4709e-01, -2.1183e-01]],\n",
       "\n",
       "         [[ 8.9816e-02, -2.5269e-01,  3.0671e-01,  ...,  1.8314e+00,\n",
       "            1.5133e+00,  3.2209e-01],\n",
       "          [-2.1556e-01,  2.3232e+00, -2.4717e-01,  ...,  2.4957e+00,\n",
       "           -8.3741e-01,  9.6921e-02],\n",
       "          [-9.8338e-01,  2.0204e+00,  4.3598e-01,  ...,  2.1060e+00,\n",
       "           -6.6111e-01, -2.4630e-02],\n",
       "          ...,\n",
       "          [-5.6778e-01,  1.1310e+00, -1.2748e-01,  ...,  6.5019e-01,\n",
       "           -1.5915e+00, -5.5325e-01],\n",
       "          [-3.1825e-01,  4.0737e-01, -2.0020e-01,  ...,  1.0246e-01,\n",
       "           -3.9801e-01,  5.2646e-01],\n",
       "          [ 6.1589e-01, -4.5050e-02, -2.8514e-01,  ...,  2.8159e-01,\n",
       "           -3.6069e-02, -6.4275e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-3.9262e-01,  1.9201e+00, -2.9125e-01,  ...,  1.2323e+00,\n",
       "            1.5457e+00, -1.4399e-01],\n",
       "          [ 1.5933e+00, -7.0192e-02,  6.4032e-01,  ...,  4.4684e-01,\n",
       "            2.9496e-01,  3.9247e-01],\n",
       "          [ 1.5230e-01, -6.0408e-01,  5.6071e-01,  ...,  9.1781e-01,\n",
       "            1.6642e+00,  5.5050e-01],\n",
       "          ...,\n",
       "          [-4.9807e-01, -6.1527e-01,  8.6227e-01,  ...,  8.9467e-01,\n",
       "            2.9639e-02,  1.9296e-01],\n",
       "          [ 6.3271e-01,  1.7760e-01,  5.3420e-01,  ..., -2.4473e-01,\n",
       "            1.4469e+00, -8.5596e-02],\n",
       "          [-7.0910e-01,  3.7817e-01,  1.4170e-03,  ...,  6.4799e-01,\n",
       "            8.7378e-01, -4.6160e-01]],\n",
       "\n",
       "         [[-5.1562e-01,  9.7841e-01,  5.1739e-02,  ..., -7.3043e-01,\n",
       "            9.0686e-01,  3.5925e-01],\n",
       "          [ 1.8866e-01, -4.9137e-01, -9.1595e-02,  ...,  3.4471e-01,\n",
       "           -2.8667e-02, -5.5394e-01],\n",
       "          [ 6.9521e-01,  2.1748e-01, -4.9266e-01,  ...,  1.1161e-01,\n",
       "            1.9028e+00,  1.6279e+00],\n",
       "          ...,\n",
       "          [-3.8466e-01,  8.2613e-01,  1.3961e+00,  ..., -2.7625e-01,\n",
       "            6.1409e-01, -3.8766e-01],\n",
       "          [-1.7703e-01,  4.7305e-02,  4.2332e-01,  ..., -2.4313e-01,\n",
       "            2.2468e-01,  5.5175e-01],\n",
       "          [ 5.5410e-02,  2.4820e-01,  1.3070e-02,  ..., -8.3069e-01,\n",
       "            1.3541e-01,  6.3415e-01]],\n",
       "\n",
       "         [[-1.1033e+00,  1.1924e+00,  2.2244e-01,  ...,  8.5723e-01,\n",
       "           -1.4499e+00, -1.7812e-01],\n",
       "          [-1.6300e-01,  5.8013e-01,  5.1730e-01,  ..., -2.9405e-01,\n",
       "           -8.0490e-01, -1.1108e+00],\n",
       "          [ 7.6437e-01, -1.8566e+00, -2.0490e-01,  ...,  6.7410e-01,\n",
       "           -6.2002e-01, -1.5637e+00],\n",
       "          ...,\n",
       "          [ 1.2301e+00, -6.5454e-01,  3.4114e-01,  ..., -8.4104e-01,\n",
       "           -2.7621e-01, -1.0421e+00],\n",
       "          [-1.5466e-01, -9.0664e-01,  5.0816e-01,  ..., -1.9881e-01,\n",
       "           -6.4918e-02, -7.2639e-01],\n",
       "          [ 6.1696e-01,  6.9433e-02, -1.1857e-01,  ...,  3.2336e-01,\n",
       "           -1.7820e-01,  2.2278e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-1.0135e-01,  5.0322e-01, -8.3144e-01,  ...,  7.0799e-01,\n",
       "            9.4913e-01,  8.4661e-01],\n",
       "          [ 5.3230e-01,  2.5842e-01, -5.6362e-01,  ...,  4.5092e-01,\n",
       "            6.1863e-01,  4.7292e-01],\n",
       "          [ 5.7432e-01,  1.0164e-01, -5.4443e-01,  ...,  4.6256e-01,\n",
       "            4.4986e-01,  1.1007e-01],\n",
       "          ...,\n",
       "          [ 2.4963e-01,  2.2949e-01, -2.6575e-01,  ...,  7.3059e-01,\n",
       "            8.5951e-01, -1.3092e-02],\n",
       "          [ 2.0730e-01, -5.6311e-02, -4.7726e-01,  ...,  7.8022e-01,\n",
       "            8.0630e-01,  1.4754e-02],\n",
       "          [-2.6240e-01,  4.4489e-01, -2.1046e-01,  ...,  4.8796e-01,\n",
       "            1.0296e+00,  2.8634e-02]],\n",
       "\n",
       "         [[-5.4964e-01, -4.3226e-02, -1.2511e-01,  ..., -1.1941e+00,\n",
       "           -3.5201e-01,  3.0434e-01],\n",
       "          [ 2.3870e-02, -7.2488e-02, -1.9536e-01,  ..., -9.1284e-01,\n",
       "            1.1092e-01,  1.2259e-01],\n",
       "          [ 1.7828e-01, -2.4147e-01,  1.6393e-01,  ..., -7.4000e-01,\n",
       "            1.3497e-01, -3.3639e-02],\n",
       "          ...,\n",
       "          [ 4.2285e-01, -2.2033e-01,  1.5717e-01,  ..., -6.1636e-01,\n",
       "            3.3296e-01,  3.5807e-01],\n",
       "          [ 1.1503e-01, -3.4745e-01, -2.2741e-01,  ..., -7.0733e-01,\n",
       "           -3.0128e-02,  1.3316e-01],\n",
       "          [-2.6329e-01, -2.3150e-01, -6.4355e-02,  ..., -4.4009e-01,\n",
       "            4.8827e-02,  3.0446e-01]],\n",
       "\n",
       "         [[ 1.4264e+00,  5.5534e-01,  6.3682e-02,  ..., -1.1099e-01,\n",
       "            3.5793e-01, -3.7667e-03],\n",
       "          [ 1.0314e+00,  9.1603e-02,  3.2919e-01,  ...,  4.1001e-01,\n",
       "            3.2279e-01, -2.4897e-01],\n",
       "          [ 5.8439e-01, -6.3257e-02,  3.4301e-01,  ...,  3.8706e-01,\n",
       "            3.1974e-01, -4.2582e-01],\n",
       "          ...,\n",
       "          [ 6.8516e-01,  2.8342e-01,  2.5720e-01,  ...,  3.3338e-01,\n",
       "            9.4524e-02, -2.9065e-01],\n",
       "          [ 7.1608e-01,  4.1526e-01, -4.0510e-02,  ...,  2.8175e-01,\n",
       "            1.8626e-01, -6.0267e-01],\n",
       "          [ 8.4218e-01,  4.0321e-01,  7.1607e-01,  ..., -4.3964e-02,\n",
       "            4.7603e-01, -3.3547e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-4.4796e-01, -3.2648e-01, -1.9400e-01,  ...,  3.3636e-01,\n",
       "           -8.0022e-01, -4.9200e-01],\n",
       "          [-2.2417e-01, -2.6312e-01, -2.8417e-02,  ...,  5.1188e-01,\n",
       "           -3.2948e-01, -1.9527e-01],\n",
       "          [-2.2902e-01,  1.8902e-02, -2.5193e-02,  ...,  3.2651e-01,\n",
       "           -1.3422e-01,  1.1619e-01],\n",
       "          ...,\n",
       "          [ 1.0643e-01,  3.1768e-03, -1.5782e-01,  ...,  2.0770e-01,\n",
       "           -3.2122e-01, -3.0614e-01],\n",
       "          [-4.8674e-01, -1.8742e-01, -2.1088e-01,  ...,  4.7069e-01,\n",
       "           -5.5035e-01, -4.4262e-01],\n",
       "          [-3.1042e-01, -3.6710e-01, -1.6290e-01,  ...,  5.3978e-01,\n",
       "           -4.9984e-01, -3.6406e-01]],\n",
       "\n",
       "         [[ 8.9584e-01, -2.6573e-01,  4.9746e-01,  ...,  1.3807e-01,\n",
       "           -7.9019e-01,  3.3373e-01],\n",
       "          [ 4.0714e-01, -3.2216e-01,  9.2660e-01,  ..., -2.8229e-01,\n",
       "            8.6175e-03,  1.3113e-01],\n",
       "          [ 6.4786e-01, -9.8481e-02,  7.8708e-01,  ...,  2.7903e-02,\n",
       "            1.7139e-01,  1.9020e-01],\n",
       "          ...,\n",
       "          [ 4.1725e-01, -4.2408e-01,  1.0042e+00,  ..., -1.9643e-02,\n",
       "           -1.5780e-01, -1.5229e-01],\n",
       "          [ 7.4871e-01, -3.8049e-01,  7.1345e-01,  ...,  9.2292e-02,\n",
       "            1.4936e-03,  8.4485e-02],\n",
       "          [ 7.4071e-01, -6.1198e-01, -1.7959e-01,  ...,  3.2645e-01,\n",
       "           -1.5954e-01, -7.7202e-02]],\n",
       "\n",
       "         [[ 7.6513e-01, -2.3269e-01,  2.5329e-01,  ..., -5.0920e-01,\n",
       "            7.5241e-02,  6.7327e-02],\n",
       "          [ 1.2825e-01, -3.6250e-01,  7.4463e-02,  ..., -4.5413e-01,\n",
       "            2.8864e-01,  4.7481e-01],\n",
       "          [ 4.0466e-01, -4.7049e-01,  2.1738e-01,  ..., -8.5108e-01,\n",
       "            4.3258e-01,  2.9862e-01],\n",
       "          ...,\n",
       "          [ 1.6059e-01, -2.1987e-01, -4.1521e-02,  ..., -6.5180e-01,\n",
       "            3.3622e-01,  1.9698e-01],\n",
       "          [ 2.3320e-01, -3.2209e-01,  9.0441e-02,  ..., -4.2901e-01,\n",
       "            1.9264e-01,  7.6602e-02],\n",
       "          [ 6.9287e-01, -7.1073e-01,  2.0437e-01,  ..., -9.4306e-01,\n",
       "           -8.9984e-02,  3.7719e-01]]],\n",
       "\n",
       "\n",
       "        [[[-8.8336e-02,  5.0057e-01, -8.3005e-01,  ...,  6.9126e-01,\n",
       "            9.6372e-01,  8.4942e-01],\n",
       "          [ 5.8787e-01,  5.5824e-01, -6.5100e-01,  ...,  8.8470e-01,\n",
       "            9.5557e-01,  4.3215e-01],\n",
       "          [ 5.3107e-01,  2.0812e-01, -9.8753e-01,  ...,  5.3796e-01,\n",
       "            7.5155e-01,  1.6339e-01],\n",
       "          ...,\n",
       "          [ 1.6138e-02,  3.1512e-01, -1.9815e-01,  ...,  5.4708e-01,\n",
       "            9.1574e-01,  3.1509e-01],\n",
       "          [ 3.6237e-01,  4.5310e-02, -4.3942e-01,  ...,  7.2004e-01,\n",
       "            7.4730e-01, -4.8211e-02],\n",
       "          [ 1.7313e-01,  2.0217e-01, -1.4031e-01,  ...,  8.1952e-01,\n",
       "            7.9490e-01, -1.1541e-01]],\n",
       "\n",
       "         [[-5.9749e-01, -3.5482e-02, -9.3377e-02,  ..., -1.1880e+00,\n",
       "           -3.3901e-01,  3.1567e-01],\n",
       "          [ 1.5503e-01, -8.2530e-02, -2.6358e-01,  ..., -7.2562e-01,\n",
       "           -4.0404e-01, -9.6563e-02],\n",
       "          [-1.5346e-01, -2.9449e-01, -2.0951e-03,  ..., -9.6485e-01,\n",
       "           -1.7721e-01,  2.7600e-01],\n",
       "          ...,\n",
       "          [-4.2255e-02, -4.7959e-01,  6.2607e-02,  ..., -8.7548e-01,\n",
       "            9.6116e-02,  2.0467e-01],\n",
       "          [ 1.9541e-01, -4.1067e-02, -3.4496e-03,  ..., -5.6414e-01,\n",
       "            3.6671e-01,  2.7185e-01],\n",
       "          [-4.1808e-02,  3.3509e-02, -1.6766e-01,  ..., -4.2874e-01,\n",
       "            6.1041e-02,  2.5154e-01]],\n",
       "\n",
       "         [[ 1.4357e+00,  5.1399e-01,  7.0892e-02,  ..., -1.0905e-01,\n",
       "            3.5672e-01, -1.5362e-02],\n",
       "          [ 1.0160e+00,  4.9997e-01,  2.0703e-01,  ...,  2.2192e-01,\n",
       "            1.9761e-01, -1.7308e-01],\n",
       "          [ 6.5464e-01,  2.7240e-01,  2.3524e-01,  ...,  2.2959e-01,\n",
       "            2.1166e-01, -5.0908e-01],\n",
       "          ...,\n",
       "          [ 6.4730e-01,  1.2457e-01,  4.6673e-01,  ...,  4.2887e-01,\n",
       "            5.9108e-02, -6.7076e-01],\n",
       "          [ 6.2084e-01,  4.0437e-02,  3.1011e-01,  ...,  5.6312e-01,\n",
       "            1.9329e-01, -5.2502e-01],\n",
       "          [ 5.6455e-01,  4.1585e-01,  1.5321e-01,  ...,  2.3929e-01,\n",
       "            3.3808e-01, -1.6230e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-4.3333e-01, -3.2085e-01, -1.9814e-01,  ...,  3.2215e-01,\n",
       "           -8.4365e-01, -5.0333e-01],\n",
       "          [-2.1394e-01, -2.7975e-02,  2.3157e-01,  ...,  4.7847e-01,\n",
       "           -2.3074e-01, -8.2352e-02],\n",
       "          [-2.9996e-01, -5.6281e-02,  2.9960e-02,  ...,  3.3428e-01,\n",
       "           -5.2395e-01, -1.0849e-01],\n",
       "          ...,\n",
       "          [ 3.7082e-01, -4.6668e-02, -3.6802e-01,  ...,  1.9155e-01,\n",
       "           -3.2066e-01,  1.5477e-01],\n",
       "          [-1.6761e-01, -2.1781e-02,  1.7648e-02,  ...,  2.0868e-01,\n",
       "           -7.7922e-01, -2.3126e-01],\n",
       "          [-5.3741e-02, -3.0489e-01, -1.9199e-01,  ...,  2.2060e-01,\n",
       "           -6.4049e-01, -2.5981e-01]],\n",
       "\n",
       "         [[ 9.0350e-01, -2.7439e-01,  5.1565e-01,  ...,  1.1033e-01,\n",
       "           -7.7389e-01,  3.0576e-01],\n",
       "          [ 6.9511e-01, -4.8989e-01,  6.4852e-01,  ...,  1.6667e-01,\n",
       "           -1.3003e-01, -1.1053e-01],\n",
       "          [ 8.7289e-01, -4.0185e-01,  9.6145e-01,  ..., -9.5136e-02,\n",
       "           -1.1511e-01, -2.1947e-02],\n",
       "          ...,\n",
       "          [ 5.6912e-01, -2.7346e-01,  5.9079e-01,  ...,  3.5938e-02,\n",
       "           -2.3780e-01, -1.3839e-01],\n",
       "          [ 4.9568e-01, -1.1558e-01,  9.4700e-01,  ..., -2.1954e-02,\n",
       "           -1.4401e-01,  7.8245e-02],\n",
       "          [ 7.6327e-01, -1.7541e-01,  2.6433e-01,  ..., -1.3924e-02,\n",
       "            1.1366e-01,  2.8212e-01]],\n",
       "\n",
       "         [[ 7.8060e-01, -2.5823e-01,  2.7373e-01,  ..., -4.8063e-01,\n",
       "            4.9384e-02,  6.3148e-02],\n",
       "          [ 9.1961e-02, -1.8831e-01,  5.4235e-02,  ..., -6.1856e-01,\n",
       "           -3.4085e-02,  4.8661e-01],\n",
       "          [ 3.2913e-01, -2.7388e-01,  2.1998e-02,  ..., -5.1950e-01,\n",
       "            4.1902e-01,  2.1131e-02],\n",
       "          ...,\n",
       "          [ 2.9559e-01, -4.3060e-01,  1.0940e-01,  ..., -6.8482e-01,\n",
       "            6.4942e-02,  7.1385e-03],\n",
       "          [ 3.3928e-01, -2.0280e-01,  1.6795e-01,  ..., -6.1804e-01,\n",
       "            1.6666e-01,  1.8755e-01],\n",
       "          [ 4.3325e-01, -2.5422e-01,  1.6275e-01,  ..., -6.0941e-01,\n",
       "           -1.5855e-04,  3.4902e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.1329e-01,  4.6682e-01, -8.1706e-01,  ...,  6.9158e-01,\n",
       "            9.4253e-01,  8.5518e-01],\n",
       "          [ 1.4050e-01,  1.5764e-02, -8.9503e-01,  ...,  4.9724e-01,\n",
       "            9.6857e-01,  1.2734e-01],\n",
       "          [ 2.3827e-01,  7.8009e-02, -5.4150e-01,  ...,  6.2665e-01,\n",
       "            7.0524e-01, -1.3811e-01],\n",
       "          ...,\n",
       "          [ 3.3709e-02,  2.6843e-01, -3.9906e-01,  ...,  6.6628e-01,\n",
       "            1.0716e+00,  6.0545e-01],\n",
       "          [ 2.0695e-01,  1.8518e-01, -4.3208e-01,  ...,  8.1054e-01,\n",
       "            6.1709e-01,  4.9522e-02],\n",
       "          [ 1.6876e-01,  1.7402e-01, -1.1832e-02,  ...,  4.9050e-01,\n",
       "            9.3803e-01,  1.5285e-01]],\n",
       "\n",
       "         [[-5.7530e-01, -4.0468e-02, -9.6236e-02,  ..., -1.1651e+00,\n",
       "           -3.2455e-01,  2.9932e-01],\n",
       "          [-1.4712e-01, -4.6738e-01, -2.9733e-01,  ..., -7.1236e-01,\n",
       "           -4.6701e-02,  4.0989e-03],\n",
       "          [-1.3532e-01, -5.4504e-01, -1.9326e-01,  ..., -5.2420e-01,\n",
       "           -8.4621e-02,  5.4548e-02],\n",
       "          ...,\n",
       "          [ 3.4741e-02, -3.2084e-01, -2.1435e-02,  ..., -5.4003e-01,\n",
       "            1.8054e-01,  2.4842e-01],\n",
       "          [ 2.2776e-01, -4.4886e-01, -5.3422e-02,  ..., -6.9334e-01,\n",
       "           -4.0998e-02, -6.4517e-03],\n",
       "          [-1.3331e-01,  9.0092e-03, -2.4612e-01,  ..., -5.3577e-01,\n",
       "            1.4306e-01,  4.1284e-01]],\n",
       "\n",
       "         [[ 1.4231e+00,  5.1060e-01,  8.1173e-02,  ..., -9.5241e-02,\n",
       "            3.5946e-01, -2.6826e-02],\n",
       "          [ 8.9070e-01,  1.0643e-01,  2.2147e-01,  ...,  2.9029e-01,\n",
       "            2.7810e-01, -4.5784e-01],\n",
       "          [ 6.8209e-01,  2.5359e-01,  1.6325e-01,  ...,  2.5936e-01,\n",
       "            1.5236e-01, -3.0699e-01],\n",
       "          ...,\n",
       "          [ 5.9673e-01,  3.6906e-01,  5.1472e-01,  ...,  2.8508e-01,\n",
       "            4.4363e-01, -2.5602e-01],\n",
       "          [ 8.6703e-01,  3.1862e-01,  7.7212e-02,  ...,  4.7820e-02,\n",
       "            7.7538e-02, -2.4955e-01],\n",
       "          [ 8.3201e-01,  2.4072e-01,  2.6890e-01,  ...,  3.0541e-01,\n",
       "            4.8360e-01, -1.6019e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-4.3891e-01, -3.3067e-01, -1.9523e-01,  ...,  3.3257e-01,\n",
       "           -8.3233e-01, -4.9170e-01],\n",
       "          [-3.2598e-01, -3.0442e-01, -3.5434e-02,  ...,  3.5961e-01,\n",
       "           -5.5506e-01, -5.0784e-01],\n",
       "          [-2.9484e-01, -2.7301e-01, -1.2343e-01,  ...,  2.6238e-01,\n",
       "           -6.8532e-01, -5.4451e-01],\n",
       "          ...,\n",
       "          [-2.8800e-01,  7.5243e-02, -2.1713e-01,  ...,  6.0300e-01,\n",
       "           -5.1002e-01, -1.3039e-01],\n",
       "          [-3.1284e-01, -2.5323e-01, -9.5432e-02,  ...,  3.5283e-01,\n",
       "           -6.6500e-01, -2.2062e-01],\n",
       "          [-1.1844e-01, -2.4189e-01,  4.4301e-02,  ...,  3.3670e-01,\n",
       "           -5.0197e-01, -3.8961e-02]],\n",
       "\n",
       "         [[ 8.8586e-01, -2.7358e-01,  5.0159e-01,  ...,  1.2748e-01,\n",
       "           -7.4753e-01,  2.9202e-01],\n",
       "          [ 5.8713e-01, -3.9235e-01,  6.2510e-01,  ..., -4.7226e-02,\n",
       "           -2.5316e-01, -2.1920e-01],\n",
       "          [ 7.4757e-01, -2.1277e-01,  6.8364e-01,  ...,  2.8920e-01,\n",
       "           -1.5120e-01, -1.4854e-01],\n",
       "          ...,\n",
       "          [ 6.7138e-01, -4.1744e-01,  6.8901e-01,  ...,  8.6316e-03,\n",
       "            3.1571e-02, -5.7384e-02],\n",
       "          [ 7.4184e-01, -3.6432e-01,  5.4372e-01,  ..., -8.2298e-02,\n",
       "           -2.5582e-01,  4.3855e-01],\n",
       "          [ 9.0798e-01, -3.3445e-01, -1.4112e-02,  ...,  1.8006e-01,\n",
       "            1.8854e-02, -2.6535e-02]],\n",
       "\n",
       "         [[ 7.9631e-01, -2.6280e-01,  2.8211e-01,  ..., -4.8469e-01,\n",
       "            5.6612e-02,  7.0603e-02],\n",
       "          [ 4.9496e-01, -2.6782e-01,  3.2656e-01,  ..., -3.4652e-01,\n",
       "           -1.1049e-02,  1.8278e-03],\n",
       "          [ 4.1101e-01, -6.0680e-01,  4.0994e-01,  ..., -5.3976e-01,\n",
       "            3.1312e-01,  1.4248e-01],\n",
       "          ...,\n",
       "          [ 5.4064e-01, -1.9087e-01,  3.3542e-01,  ..., -4.0016e-01,\n",
       "            2.8365e-01,  9.7889e-02],\n",
       "          [ 4.6157e-01, -3.9022e-01,  1.0111e-01,  ..., -7.0636e-01,\n",
       "            3.2491e-01,  1.7425e-01],\n",
       "          [ 3.8891e-01, -5.2672e-01,  1.7049e-01,  ..., -4.5040e-01,\n",
       "           -3.5702e-01,  3.9696e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 0.0676, -0.1759, -0.1530,  ..., -0.1035, -0.1043,  0.1499],\n",
       "          [-0.0279, -0.1212,  0.1118,  ..., -0.3991, -0.0083, -0.0230],\n",
       "          [-0.0108,  0.0654,  0.0649,  ..., -0.4001, -0.0626, -0.2318],\n",
       "          ...,\n",
       "          [-0.2579, -0.3120,  0.2414,  ..., -0.1867, -0.1986, -0.1719],\n",
       "          [-0.0378, -0.1610,  0.0815,  ..., -0.3997, -0.2555, -0.1372],\n",
       "          [-0.3308, -0.5656,  0.4528,  ..., -0.3962, -0.4563, -0.3705]],\n",
       "\n",
       "         [[ 0.2625,  0.1595,  1.1794,  ..., -0.4090,  0.5785,  0.4939],\n",
       "          [ 0.1800,  0.3518,  1.0985,  ...,  0.1022,  0.5313,  0.1466],\n",
       "          [ 0.2753,  0.0130,  1.0670,  ..., -0.0684,  0.2575,  0.1795],\n",
       "          ...,\n",
       "          [-0.0980,  0.3144,  1.1810,  ...,  0.3846,  0.2824,  0.2953],\n",
       "          [ 0.0787,  0.3334,  1.0967,  ...,  0.0307,  0.5810,  0.1059],\n",
       "          [ 0.4300,  0.4752,  1.2854,  ...,  0.0405,  0.1588,  0.2167]],\n",
       "\n",
       "         [[ 0.2549, -0.9919, -0.0051,  ...,  0.7422,  0.0101, -0.9563],\n",
       "          [ 0.0961, -1.0744,  0.0286,  ...,  0.9202,  0.0191, -0.3319],\n",
       "          [ 0.2879, -1.1203, -0.0620,  ...,  0.6116, -0.0771, -0.4704],\n",
       "          ...,\n",
       "          [-0.0348, -1.5353,  0.1068,  ...,  0.3628,  0.1949,  0.0461],\n",
       "          [ 0.1113, -1.3386,  0.0228,  ...,  0.5707,  0.2209, -0.6717],\n",
       "          [ 0.7062, -1.1033, -0.0092,  ...,  0.6926,  0.2577, -0.4916]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 0.2911,  0.8647,  0.2183,  ..., -0.0231,  0.4839,  0.0243],\n",
       "          [ 0.4763,  1.0525, -0.1170,  ...,  0.2437,  0.2698, -0.1468],\n",
       "          [ 0.3843,  1.0893, -0.0065,  ...,  0.1441, -0.0973, -0.1834],\n",
       "          ...,\n",
       "          [ 0.3036,  0.6188,  0.3197,  ...,  0.0767,  0.0452, -0.0676],\n",
       "          [ 0.4699,  0.9968,  0.0483,  ...,  0.1337,  0.1083,  0.0551],\n",
       "          [ 0.8705,  0.1815,  0.0280,  ...,  0.6830, -0.2091, -0.0450]],\n",
       "\n",
       "         [[-0.3822, -0.5888, -0.1259,  ...,  0.0540, -0.1490,  0.3461],\n",
       "          [-0.6253, -0.5512,  0.0767,  ..., -0.0076,  0.0929, -0.1688],\n",
       "          [-0.7044, -0.6114,  0.0107,  ...,  0.0583, -0.1261, -0.1364],\n",
       "          ...,\n",
       "          [-0.3969, -0.4676, -0.0074,  ...,  0.2521, -0.2618,  0.1204],\n",
       "          [-0.5154, -0.6170,  0.0725,  ...,  0.0987, -0.1234,  0.1405],\n",
       "          [-0.9309, -0.7178, -0.2077,  ...,  0.1089,  0.0333, -0.5822]],\n",
       "\n",
       "         [[-0.0022,  0.5461, -0.2111,  ..., -0.2793,  0.6060,  0.4860],\n",
       "          [-0.0343,  0.0673, -0.4686,  ...,  0.1519,  0.4749,  0.5590],\n",
       "          [-0.0410,  0.0529, -0.6518,  ..., -0.0751,  0.5849,  0.3810],\n",
       "          ...,\n",
       "          [ 0.0778,  0.3322, -0.3116,  ...,  0.0384,  0.6410,  0.0161],\n",
       "          [-0.1513,  0.5173, -0.4145,  ..., -0.2639,  0.6532,  0.2570],\n",
       "          [-0.1722,  0.4748, -0.8778,  ..., -0.2879,  0.3693,  0.9599]]],\n",
       "\n",
       "\n",
       "        [[[ 0.0592, -0.1317, -0.1645,  ..., -0.0955, -0.0632,  0.1722],\n",
       "          [ 0.1214,  0.0280,  0.0632,  ..., -0.4153,  0.0450, -0.1913],\n",
       "          [ 0.0878, -0.2304,  0.2668,  ..., -0.1698, -0.0606, -0.2357],\n",
       "          ...,\n",
       "          [ 0.0025, -0.4852, -0.0620,  ..., -0.0793,  0.1240,  0.0258],\n",
       "          [-0.0728, -0.0955,  0.1468,  ..., -0.3288, -0.2832, -0.0707],\n",
       "          [-0.0766, -0.4278,  0.2783,  ..., -0.4788, -0.2446, -0.2384]],\n",
       "\n",
       "         [[ 0.2335,  0.1437,  1.1820,  ..., -0.3781,  0.5975,  0.5097],\n",
       "          [ 0.5581,  0.2174,  1.7225,  ...,  0.5710,  0.8460,  0.1094],\n",
       "          [ 0.2488,  0.0783,  1.2853,  ...,  0.0669,  0.4872,  0.1599],\n",
       "          ...,\n",
       "          [ 0.1472,  0.2368,  1.4495,  ...,  0.6304,  0.4332, -0.0067],\n",
       "          [-0.0202,  0.2814,  1.4234,  ...,  0.1037,  0.4701,  0.4324],\n",
       "          [ 0.1158,  0.2001,  1.4674,  ...,  0.1057,  0.2456,  0.4657]],\n",
       "\n",
       "         [[ 0.2773, -1.0344,  0.0120,  ...,  0.7679, -0.0113, -0.9370],\n",
       "          [ 0.1426, -1.5701,  0.4417,  ...,  0.9969,  0.1351, -0.4915],\n",
       "          [-0.1252, -1.1818,  0.0991,  ...,  0.5870, -0.0548, -0.5904],\n",
       "          ...,\n",
       "          [ 0.4971, -1.7400,  0.1580,  ...,  0.7459,  0.2655, -0.3805],\n",
       "          [ 0.1978, -1.2355, -0.2391,  ...,  0.4852,  0.2101, -0.5394],\n",
       "          [ 0.2082, -1.1453, -0.1075,  ...,  0.7423, -0.0473, -0.2582]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 0.3112,  0.8238,  0.1721,  ..., -0.0505,  0.4747,  0.0590],\n",
       "          [ 0.6463,  0.6235, -0.1442,  ...,  0.1469, -0.0457,  0.1714],\n",
       "          [ 0.4433,  0.7898,  0.2503,  ..., -0.0458, -0.0190,  0.0876],\n",
       "          ...,\n",
       "          [ 0.7066,  0.2959,  0.2009,  ..., -0.1753, -0.1361,  0.3448],\n",
       "          [ 0.0788,  0.8024,  0.1075,  ..., -0.1376,  0.1506, -0.0434],\n",
       "          [ 0.6469,  0.6352, -0.0323,  ...,  0.2149, -0.0215, -0.0176]],\n",
       "\n",
       "         [[-0.3588, -0.5816, -0.1106,  ...,  0.0625, -0.1874,  0.3267],\n",
       "          [-0.5965, -0.3680, -0.2023,  ...,  0.2204, -0.2254, -0.0649],\n",
       "          [-0.5270, -0.4117, -0.0500,  ...,  0.1166, -0.1513,  0.0380],\n",
       "          ...,\n",
       "          [-0.4226, -0.2866,  0.0702,  ...,  0.2605, -0.4704, -0.2377],\n",
       "          [-0.4506, -0.5437,  0.1213,  ..., -0.0834, -0.2368, -0.1735],\n",
       "          [-0.3690, -0.7059, -0.1450,  ...,  0.3699, -0.1054, -0.2116]],\n",
       "\n",
       "         [[-0.0021,  0.5435, -0.2019,  ..., -0.2577,  0.5926,  0.4571],\n",
       "          [ 0.0812,  0.5484, -0.5849,  ...,  0.3671,  0.5107,  0.3108],\n",
       "          [ 0.1055,  0.4434, -0.5033,  ..., -0.2228,  0.5945,  0.4403],\n",
       "          ...,\n",
       "          [ 0.0970,  0.5545, -0.3111,  ...,  0.2830,  0.9005, -0.0992],\n",
       "          [-0.0189,  0.4231, -0.1763,  ...,  0.0762,  0.5249, -0.0118],\n",
       "          [-0.4656,  0.2447, -0.4981,  ..., -0.2325,  0.7413,  0.4197]]],\n",
       "\n",
       "\n",
       "        [[[ 0.0611, -0.1501, -0.1612,  ..., -0.0940, -0.0549,  0.1517],\n",
       "          [ 0.0773, -0.1937, -0.0627,  ..., -0.1253,  0.2634, -0.1106],\n",
       "          [-0.0360, -0.2564,  0.1483,  ..., -0.3542,  0.0300, -0.0514],\n",
       "          ...,\n",
       "          [ 0.2436, -0.4214,  0.1529,  ..., -0.2810, -0.0820, -0.2381],\n",
       "          [ 0.0416, -0.3442,  0.0382,  ..., -0.4389, -0.0860, -0.2190],\n",
       "          [-0.0795, -0.6417,  0.2901,  ..., -0.3639, -0.2454, -0.2957]],\n",
       "\n",
       "         [[ 0.2514,  0.1365,  1.1843,  ..., -0.3739,  0.5792,  0.4999],\n",
       "          [-0.1125,  0.2204,  0.8767,  ..., -0.1214,  0.4892,  0.1039],\n",
       "          [ 0.0446,  0.0550,  1.1072,  ...,  0.0343,  0.4551,  0.0957],\n",
       "          ...,\n",
       "          [ 0.1981,  0.4457,  1.1187,  ...,  0.2610,  0.5551,  0.0287],\n",
       "          [-0.0304,  0.1615,  1.2859,  ...,  0.1874,  0.2371,  0.2761],\n",
       "          [ 0.1151,  0.3768,  1.0068,  ...,  0.1196,  0.6153,  0.1766]],\n",
       "\n",
       "         [[ 0.2628, -1.0371,  0.0045,  ...,  0.7758, -0.0118, -0.9487],\n",
       "          [-0.0875, -1.1440, -0.3602,  ...,  0.7061, -0.1133, -0.6912],\n",
       "          [-0.0131, -1.0721, -0.2704,  ...,  0.6245, -0.0974, -0.9119],\n",
       "          ...,\n",
       "          [ 0.2994, -1.3022, -0.0560,  ...,  0.5795,  0.2126, -0.4197],\n",
       "          [ 0.0612, -1.2138, -0.1399,  ...,  0.8073,  0.0773, -0.8443],\n",
       "          [ 0.3558, -1.1413,  0.1858,  ...,  0.7171, -0.0463, -0.0494]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 0.3026,  0.8477,  0.1816,  ..., -0.0272,  0.4838,  0.0339],\n",
       "          [ 0.2294,  0.8880,  0.0625,  ...,  0.0893,  0.0670,  0.3101],\n",
       "          [ 0.5679,  0.9863,  0.0653,  ...,  0.1791,  0.2979,  0.0860],\n",
       "          ...,\n",
       "          [ 0.2398,  0.6828,  0.3288,  ...,  0.3994, -0.1279,  0.0413],\n",
       "          [ 0.5770,  1.1119,  0.0836,  ...,  0.0486,  0.3047, -0.0332],\n",
       "          [ 0.5016,  0.5659,  0.1277,  ...,  0.3406, -0.2308, -0.1972]],\n",
       "\n",
       "         [[-0.3922, -0.5915, -0.1145,  ...,  0.0585, -0.1818,  0.3300],\n",
       "          [-0.5517, -0.6992,  0.2283,  ..., -0.1368, -0.3024,  0.2486],\n",
       "          [-0.4628, -0.7165,  0.1869,  ...,  0.0018, -0.2727,  0.1856],\n",
       "          ...,\n",
       "          [-0.6663, -0.7946,  0.2994,  ..., -0.1306,  0.0109,  0.3921],\n",
       "          [-0.4130, -0.8766, -0.0429,  ...,  0.1809, -0.1075, -0.0299],\n",
       "          [-0.7184, -0.7108, -0.0696,  ...,  0.7130,  0.0030, -0.1231]],\n",
       "\n",
       "         [[-0.0078,  0.5398, -0.2338,  ..., -0.2451,  0.5854,  0.4705],\n",
       "          [-0.0695,  0.5367, -0.2009,  ..., -0.1732,  0.3571,  0.2420],\n",
       "          [-0.2836,  0.5005, -0.3517,  ..., -0.3176,  0.4871,  0.2910],\n",
       "          ...,\n",
       "          [ 0.0251,  0.5342, -0.3585,  ..., -0.2272,  0.4793,  0.4519],\n",
       "          [-0.1390,  0.5323, -0.4192,  ..., -0.0274,  0.6760,  0.4606],\n",
       "          [-0.2830,  0.3770, -0.5461,  ...,  0.0658,  0.6782,  0.6728]]]],\n",
       "       grad_fn=<TransposeBackward0>), tensor([[[[-2.8741e-01,  8.4516e-01, -1.5820e+00,  ..., -4.2958e-02,\n",
       "           -2.2415e-01,  6.8900e-01],\n",
       "          [-1.1926e+00, -6.2844e-01, -3.2588e-01,  ...,  2.5817e-01,\n",
       "            1.0705e+00,  2.3579e+00],\n",
       "          [ 6.1173e-01, -1.1971e+00, -9.4520e-01,  ..., -6.8775e-01,\n",
       "            7.8398e-01,  2.5654e+00],\n",
       "          ...,\n",
       "          [-4.1099e-01, -2.5097e+00, -3.5756e-01,  ...,  5.1256e-02,\n",
       "            7.6685e-01,  5.2382e-01],\n",
       "          [-1.0337e+00, -6.5883e-01,  2.4982e-01,  ..., -6.8322e-01,\n",
       "            4.1600e-01,  1.0456e+00],\n",
       "          [ 2.9875e+00, -3.1822e-01, -1.9658e+00,  ..., -1.7866e+00,\n",
       "           -2.0172e-01,  1.4852e+00]],\n",
       "\n",
       "         [[ 1.4476e+00, -1.0809e+00, -4.5413e-01,  ..., -2.5228e+00,\n",
       "           -1.3506e+00, -1.1501e+00],\n",
       "          [ 3.1102e+00,  1.8962e+00, -1.0788e+00,  ..., -1.4857e+00,\n",
       "            1.3387e+00, -2.1135e+00],\n",
       "          [ 2.5078e+00,  2.3912e-01, -1.1542e+00,  ..., -6.8677e-01,\n",
       "           -3.7409e+00,  2.0520e+00],\n",
       "          ...,\n",
       "          [ 3.0123e+00,  4.1638e-01, -8.4756e-03,  ...,  2.4126e+00,\n",
       "            1.8299e+00,  1.0264e+00],\n",
       "          [ 1.7902e+00,  1.1149e+00, -1.0554e+00,  ...,  2.3374e+00,\n",
       "           -7.0443e-01,  3.7285e+00],\n",
       "          [-1.5525e-02, -2.3380e+00, -1.5883e+00,  ...,  3.0856e+00,\n",
       "            1.0752e+00,  2.7993e+00]],\n",
       "\n",
       "         [[-8.9772e-01,  1.2256e+00,  5.9863e-01,  ...,  5.1011e+00,\n",
       "           -1.3730e+00,  2.6493e+00],\n",
       "          [ 1.5891e-01,  1.8129e+00, -1.3168e+00,  ...,  2.4750e-01,\n",
       "           -3.0200e+00,  3.3119e+00],\n",
       "          [-3.0256e-01, -5.8163e-02,  2.7093e-01,  ...,  5.9043e-01,\n",
       "           -5.3389e-01, -1.3155e+00],\n",
       "          ...,\n",
       "          [-1.8912e+00, -1.7237e+00, -3.1647e+00,  ...,  9.0730e-02,\n",
       "           -1.0518e-01,  1.4377e+00],\n",
       "          [-1.8794e-01, -9.8766e-01, -2.0078e+00,  ..., -6.5492e-01,\n",
       "            9.0505e-02, -5.0048e-01],\n",
       "          [ 3.8562e-01,  5.9551e-01, -3.4534e+00,  ..., -3.2956e-01,\n",
       "            5.6306e-01, -2.1521e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 8.5330e-02, -1.7688e+00, -2.7785e-01,  ..., -6.3620e-02,\n",
       "            1.9345e+00, -4.0010e-01],\n",
       "          [-1.9955e-01,  9.1236e-01, -6.4378e-01,  ..., -3.0592e+00,\n",
       "           -8.7070e-01, -1.1874e+00],\n",
       "          [-2.1016e-01, -2.9725e-01,  3.5105e-01,  ...,  2.0141e-01,\n",
       "           -1.3912e+00,  1.9014e-01],\n",
       "          ...,\n",
       "          [ 7.4915e-01,  5.6741e-01,  2.3872e+00,  ..., -3.3340e+00,\n",
       "           -1.3115e+00,  5.6773e-01],\n",
       "          [-1.5281e-01,  4.4044e-02, -2.6208e+00,  ..., -1.3433e+00,\n",
       "           -3.5609e+00,  1.2827e+00],\n",
       "          [-2.7910e+00,  6.4957e-01, -3.8751e-01,  ...,  8.1336e-01,\n",
       "            1.1013e+00, -3.5059e+00]],\n",
       "\n",
       "         [[-1.2447e+00,  1.8313e+00,  2.3255e+00,  ..., -3.4212e+00,\n",
       "            2.8316e-01, -6.6296e-01],\n",
       "          [-5.1236e-01, -2.4564e+00, -4.4994e-01,  ..., -2.4878e+00,\n",
       "            1.0845e+00,  3.2483e-01],\n",
       "          [-2.1730e+00, -2.1110e+00,  2.6766e+00,  ..., -2.2353e+00,\n",
       "            1.5595e+00,  1.3873e+00],\n",
       "          ...,\n",
       "          [-4.9940e-01, -9.9049e-01, -1.0842e+00,  ..., -3.5541e-01,\n",
       "            2.1101e+00, -1.5273e+00],\n",
       "          [-1.3625e+00, -5.9108e-01,  2.0228e+00,  ..., -7.3015e-01,\n",
       "            2.3232e+00,  8.7271e-01],\n",
       "          [ 2.9405e+00,  4.2526e+00, -1.3774e+00,  ..., -1.9486e+00,\n",
       "           -8.0977e-01, -2.6859e+00]],\n",
       "\n",
       "         [[ 2.3472e-01,  9.6467e-01,  8.9653e-01,  ..., -2.4161e-02,\n",
       "           -5.2153e-01, -1.1248e+00],\n",
       "          [-5.7414e-01, -8.7006e-01, -4.3004e-01,  ...,  9.5936e-02,\n",
       "           -1.7753e+00, -6.1017e-01],\n",
       "          [-9.3408e-01,  5.2229e-01, -1.8662e+00,  ...,  3.2951e-01,\n",
       "           -2.2731e+00, -3.5030e-01],\n",
       "          ...,\n",
       "          [ 8.6380e-01, -1.3475e+00, -4.4180e-01,  ...,  6.2452e-01,\n",
       "           -4.4549e-01,  1.1496e+00],\n",
       "          [-5.1059e-02,  1.0222e+00, -1.0343e+00,  ..., -1.2158e+00,\n",
       "            3.9284e-02,  9.9487e-01],\n",
       "          [ 9.8758e-03, -2.4661e-01, -3.3864e+00,  ..., -1.6057e+00,\n",
       "            3.5271e+00,  1.2255e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.1416e+00, -2.0323e-01, -1.5227e-01,  ...,  1.0047e+00,\n",
       "            1.8395e+00,  1.2606e-01],\n",
       "          [ 2.1296e+00, -1.7496e+00,  1.2252e-01,  ..., -1.3379e+00,\n",
       "            9.8792e-01,  1.4182e-01],\n",
       "          [-8.6563e-01,  2.4981e+00, -1.9173e+00,  ..., -5.4671e-02,\n",
       "            1.2005e+00, -1.4834e+00],\n",
       "          ...,\n",
       "          [ 2.1749e-01, -2.2244e+00,  4.2981e-04,  ...,  2.0111e-01,\n",
       "           -9.8568e-01,  1.8682e+00],\n",
       "          [-3.3100e-02, -1.5136e+00,  7.2680e-01,  ...,  1.8441e-01,\n",
       "           -8.2086e-01,  9.8485e-01],\n",
       "          [ 2.9807e+00, -4.2073e-01, -1.9423e+00,  ..., -1.7654e+00,\n",
       "           -1.2641e-01,  1.5153e+00]],\n",
       "\n",
       "         [[ 1.7707e+00,  7.2767e-01, -1.8143e-01,  ..., -4.0926e+00,\n",
       "            3.8252e+00, -1.5818e+00],\n",
       "          [ 2.7478e+00, -1.1524e+00, -7.7859e-01,  ..., -2.8488e-01,\n",
       "           -2.4625e+00,  4.4317e+00],\n",
       "          [ 6.5304e-01,  7.2514e-01, -5.2476e-01,  ...,  2.0516e+00,\n",
       "            3.3598e-01,  3.3080e+00],\n",
       "          ...,\n",
       "          [ 1.3385e+00, -1.9527e+00, -1.1835e+00,  ..., -5.7513e-01,\n",
       "           -2.1445e+00, -3.1903e+00],\n",
       "          [ 2.4773e+00,  1.3915e-01,  4.2605e-02,  ...,  1.0474e+00,\n",
       "           -1.9343e+00, -1.0444e+00],\n",
       "          [ 4.0115e-02, -2.3480e+00, -1.5861e+00,  ...,  3.0379e+00,\n",
       "            8.7010e-01,  2.6872e+00]],\n",
       "\n",
       "         [[-3.8298e+00,  2.2194e+00, -1.3964e+00,  ..., -1.9967e+00,\n",
       "            3.2357e-01,  1.7194e-01],\n",
       "          [-1.6649e+00, -4.2646e-01,  1.1228e+00,  ...,  2.6013e-02,\n",
       "            1.1394e-01, -3.6258e+00],\n",
       "          [ 2.0317e+00, -1.3907e+00,  1.8003e+00,  ..., -4.4650e-01,\n",
       "            2.7672e-01, -5.0908e+00],\n",
       "          ...,\n",
       "          [-1.5106e+00, -3.0773e+00, -1.4112e+00,  ...,  4.8660e+00,\n",
       "           -3.0436e+00,  1.1933e+00],\n",
       "          [-1.4014e+00, -2.9160e+00, -1.5740e+00,  ...,  1.9485e+00,\n",
       "           -3.1868e+00,  2.5831e+00],\n",
       "          [ 3.7310e-01,  5.7240e-01, -3.4513e+00,  ..., -2.7765e-01,\n",
       "            3.7785e-01, -2.1006e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.4123e+00,  1.5160e+00,  8.0984e-01,  ..., -1.4918e+00,\n",
       "           -2.7959e+00, -3.6362e+00],\n",
       "          [ 1.0079e-01,  1.1408e-01,  2.2182e-01,  ..., -8.5014e-01,\n",
       "           -2.4425e+00,  3.1106e-01],\n",
       "          [ 7.7889e-01,  1.9219e+00, -2.3196e+00,  ...,  9.3996e-02,\n",
       "            1.4420e+00, -1.4401e+00],\n",
       "          ...,\n",
       "          [ 1.9710e-01, -1.2212e+00,  2.0783e+00,  ..., -2.0370e-01,\n",
       "           -3.3980e+00, -1.0630e+00],\n",
       "          [ 2.5178e-01,  7.5574e-02,  1.5479e+00,  ..., -2.5159e+00,\n",
       "           -2.7844e+00, -5.8530e-01],\n",
       "          [-2.8031e+00,  6.4713e-01, -2.2373e-01,  ...,  7.7742e-01,\n",
       "            1.0888e+00, -3.4821e+00]],\n",
       "\n",
       "         [[ 5.6023e-01,  8.9706e-01,  7.1659e-01,  ..., -1.3415e+00,\n",
       "           -5.8466e-01, -1.7784e+00],\n",
       "          [-1.6823e+00, -2.8518e+00,  2.6048e+00,  ..., -5.8631e-01,\n",
       "            1.7061e+00,  1.3551e+00],\n",
       "          [ 1.0237e-01, -3.0982e+00,  4.1705e-01,  ..., -4.3412e-01,\n",
       "           -1.4445e+00, -2.6195e-01],\n",
       "          ...,\n",
       "          [-2.5332e+00,  4.5967e-01,  1.6214e+00,  ..., -2.6545e+00,\n",
       "            1.2224e+00, -1.0746e+00],\n",
       "          [-1.3113e+00,  8.8193e-01,  1.8058e+00,  ..., -2.3274e+00,\n",
       "            1.9234e+00, -4.1786e-01],\n",
       "          [ 2.9070e+00,  4.2666e+00, -1.3439e+00,  ..., -2.0378e+00,\n",
       "           -7.1851e-01, -2.7595e+00]],\n",
       "\n",
       "         [[-9.3329e-01, -3.4325e-02, -1.5249e-01,  ...,  5.7505e-02,\n",
       "           -1.2478e+00,  7.8611e-01],\n",
       "          [ 8.0814e-02,  9.4957e-01, -2.2746e+00,  ...,  1.8814e-01,\n",
       "           -3.2825e+00,  7.7590e-01],\n",
       "          [ 6.0835e-01,  1.1536e+00, -2.3539e+00,  ...,  1.3761e-01,\n",
       "           -4.7257e-02, -7.5997e-01],\n",
       "          ...,\n",
       "          [ 9.4537e-01,  1.8882e+00,  2.1215e+00,  ..., -9.0249e-01,\n",
       "            1.4341e+00,  1.5843e-01],\n",
       "          [-9.6843e-01, -4.7675e-03,  1.0308e+00,  ..., -1.8669e+00,\n",
       "            2.3278e+00,  1.6668e+00],\n",
       "          [-2.7913e-02, -2.4781e-01, -3.3520e+00,  ..., -1.6570e+00,\n",
       "            3.5395e+00,  1.6662e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.3191e+00, -1.6505e+00, -1.6878e+00,  ...,  1.6883e-01,\n",
       "           -8.6179e-01,  9.3030e-01],\n",
       "          [-1.3057e-01, -4.5660e-01, -1.7215e+00,  ..., -8.4962e-01,\n",
       "           -1.2490e+00, -1.3263e+00],\n",
       "          [-2.0011e+00, -2.8212e+00, -5.7443e-02,  ...,  1.5126e+00,\n",
       "           -7.1174e-01, -2.0789e+00],\n",
       "          ...,\n",
       "          [-4.0170e-01,  1.3982e+00, -4.0400e-02,  ...,  1.6630e+00,\n",
       "           -2.5169e-01, -1.5417e+00],\n",
       "          [-7.3403e-01, -1.3199e+00, -1.0155e+00,  ...,  3.3701e-01,\n",
       "            1.1459e+00, -7.5075e-01],\n",
       "          [ 2.9774e+00, -3.8437e-01, -1.9746e+00,  ..., -1.8103e+00,\n",
       "           -1.5439e-01,  1.4606e+00]],\n",
       "\n",
       "         [[ 1.5836e-01, -5.1377e-01, -1.5420e+00,  ..., -1.4686e+00,\n",
       "            4.3804e+00, -3.5273e+00],\n",
       "          [ 1.6056e+00, -1.5127e+00, -2.1337e+00,  ..., -1.2627e+00,\n",
       "            1.2438e+00, -1.6762e+00],\n",
       "          [ 1.3587e+00, -1.1531e+00, -1.1398e+00,  ...,  2.7058e+00,\n",
       "            4.0180e-01, -3.9508e-01],\n",
       "          ...,\n",
       "          [ 1.4974e+00,  2.2547e+00,  3.2460e-01,  ...,  1.2800e+00,\n",
       "           -2.7767e+00,  3.5220e-02],\n",
       "          [ 6.5102e-01,  1.1639e+00,  2.9323e-01,  ...,  7.1010e-01,\n",
       "           -1.4914e+00,  1.5577e+00],\n",
       "          [-8.2316e-02, -2.3717e+00, -1.5371e+00,  ...,  3.0215e+00,\n",
       "            9.9267e-01,  2.7367e+00]],\n",
       "\n",
       "         [[-3.1203e-01, -4.5101e-03, -1.4875e+00,  ...,  9.9255e-01,\n",
       "           -4.9843e+00, -1.9931e+00],\n",
       "          [-2.9892e+00, -1.7086e+00, -8.8849e-02,  ...,  8.8123e-02,\n",
       "           -2.0307e+00, -1.3325e+00],\n",
       "          [-1.5864e+00, -1.8375e+00, -4.6428e-01,  ..., -3.5664e+00,\n",
       "           -3.2769e+00,  5.9574e-01],\n",
       "          ...,\n",
       "          [-2.1420e+00, -1.3609e+00, -8.3650e-01,  ..., -3.7054e-02,\n",
       "           -3.6767e-01,  1.9898e+00],\n",
       "          [-1.9785e+00, -6.1620e-01, -1.5385e+00,  ...,  1.5201e+00,\n",
       "           -4.3837e-01,  1.0294e+00],\n",
       "          [ 3.7811e-01,  6.3502e-01, -3.4054e+00,  ..., -3.0708e-01,\n",
       "            4.6776e-01, -2.1774e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.8480e+00, -1.8721e+00, -1.2139e-01,  ..., -1.4457e+00,\n",
       "           -2.3449e+00, -5.0734e-02],\n",
       "          [ 1.8508e+00, -2.4385e+00, -2.4085e+00,  ..., -1.0159e+00,\n",
       "           -2.0532e+00,  3.1421e-01],\n",
       "          [ 3.1527e+00,  3.2586e+00, -7.8895e-01,  ...,  7.9543e-01,\n",
       "           -7.8542e-01,  1.5225e+00],\n",
       "          ...,\n",
       "          [ 4.0695e+00,  1.6026e+00, -1.4892e+00,  ..., -2.5871e+00,\n",
       "           -1.9690e+00,  1.2545e+00],\n",
       "          [ 7.1236e-01,  5.6706e-01, -1.2868e+00,  ..., -9.5870e-01,\n",
       "           -1.4828e+00,  1.9419e+00],\n",
       "          [-2.7961e+00,  6.7080e-01, -3.2247e-01,  ...,  7.9506e-01,\n",
       "            1.1369e+00, -3.5218e+00]],\n",
       "\n",
       "         [[-2.0683e-01, -1.0409e+00,  1.9801e+00,  ..., -1.3338e+00,\n",
       "           -2.2037e+00,  6.9567e-02],\n",
       "          [ 9.0120e-02, -5.4924e+00,  3.1490e+00,  ...,  5.6773e-02,\n",
       "           -1.0575e-01, -7.4999e-02],\n",
       "          [-1.3768e+00, -1.7999e+00,  3.1442e+00,  ...,  7.3868e-01,\n",
       "            1.7533e+00,  2.4894e+00],\n",
       "          ...,\n",
       "          [ 4.6006e-01,  1.1072e+00,  1.1900e+00,  ...,  2.2272e+00,\n",
       "            2.0794e+00,  1.7485e+00],\n",
       "          [ 1.0053e+00, -1.4643e+00,  1.5509e+00,  ..., -8.5448e-03,\n",
       "            3.5396e+00,  3.4432e-01],\n",
       "          [ 2.9291e+00,  4.2769e+00, -1.3084e+00,  ..., -1.9685e+00,\n",
       "           -8.0780e-01, -2.7022e+00]],\n",
       "\n",
       "         [[-1.0913e+00,  1.7022e+00,  2.4319e-01,  ...,  4.5010e-01,\n",
       "           -1.2825e+00, -2.1213e+00],\n",
       "          [-5.1184e-02, -1.0336e+00, -2.0554e+00,  ...,  9.3192e-01,\n",
       "            9.4659e-01, -4.3159e-01],\n",
       "          [-5.3164e-02,  1.2280e-01, -1.8926e+00,  ..., -7.4709e-01,\n",
       "           -9.4785e-01, -2.9276e+00],\n",
       "          ...,\n",
       "          [ 4.7604e-01, -1.3153e-01, -1.1712e+00,  ..., -1.7918e+00,\n",
       "            1.6863e-02,  1.5220e+00],\n",
       "          [ 6.7497e-01,  3.4883e-01, -6.8042e-01,  ..., -1.1961e+00,\n",
       "            1.3932e+00, -6.3455e-01],\n",
       "          [ 1.1499e-02, -2.3311e-01, -3.3757e+00,  ..., -1.6353e+00,\n",
       "            3.5327e+00,  1.2757e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.4712e+00, -1.0845e-01,  2.6167e-01,  ..., -5.9804e-01,\n",
       "            1.7792e-01,  9.7249e-01],\n",
       "          [-7.4056e-01,  9.2521e-01,  8.5493e-01,  ..., -3.9872e-01,\n",
       "            1.9195e+00, -4.3547e-01],\n",
       "          [-1.3625e+00,  4.9457e-01, -1.6563e-01,  ...,  1.2658e-01,\n",
       "            8.1092e-01, -5.4217e-03],\n",
       "          ...,\n",
       "          [ 2.6541e-01,  2.2301e+00, -1.3813e-01,  ..., -3.6176e-01,\n",
       "            4.0321e-01, -4.2447e-01],\n",
       "          [-4.6815e-01,  4.3084e-01, -2.7154e-01,  ..., -8.6044e-02,\n",
       "           -6.2428e-01, -5.0050e-01],\n",
       "          [ 6.3921e-01, -4.1495e-01, -2.9660e-01,  ..., -2.5774e-01,\n",
       "           -2.5850e-01,  3.4281e-01]],\n",
       "\n",
       "         [[ 1.0222e-01,  7.0568e-02,  1.6616e+00,  ..., -4.4615e-01,\n",
       "            7.9100e-01,  2.1162e+00],\n",
       "          [ 2.9309e-01,  5.4339e-01, -2.0308e-01,  ...,  5.5117e-01,\n",
       "            2.1645e-01, -1.9572e-01],\n",
       "          [-1.4548e+00,  1.3098e+00,  2.0550e-01,  ..., -1.7055e-01,\n",
       "            8.3455e-01, -1.2317e+00],\n",
       "          ...,\n",
       "          [ 4.2937e-01,  1.0121e+00, -2.7255e-01,  ...,  2.1683e-01,\n",
       "            1.0911e+00,  1.3111e+00],\n",
       "          [ 3.4428e-03,  7.9910e-01,  1.1676e+00,  ..., -1.6570e-01,\n",
       "            7.2536e-01, -5.3851e-01],\n",
       "          [-1.5720e-01,  1.4207e-01, -5.8504e-01,  ...,  5.9588e-01,\n",
       "           -1.6956e-01, -3.2404e-01]],\n",
       "\n",
       "         [[-5.5740e-01,  5.6077e-01,  2.1276e-02,  ..., -5.9591e-01,\n",
       "           -1.7737e+00,  2.7820e+00],\n",
       "          [-1.0376e+00,  7.8261e-01,  1.6148e+00,  ..., -2.2017e-02,\n",
       "           -7.2130e-01,  3.2131e+00],\n",
       "          [-2.4851e-01,  1.3756e+00,  2.1731e+00,  ..., -1.0421e+00,\n",
       "           -1.2903e-01,  1.8287e+00],\n",
       "          ...,\n",
       "          [ 4.0242e-01,  5.1936e-01,  1.2856e-01,  ...,  6.7551e-01,\n",
       "           -2.0105e+00,  3.4126e+00],\n",
       "          [ 1.2278e+00,  4.2892e-01, -4.6849e-01,  ...,  4.6698e-01,\n",
       "           -1.0225e+00,  4.5135e-01],\n",
       "          [-3.5011e-01,  4.1631e-01, -8.9937e-03,  ..., -3.7172e-01,\n",
       "           -5.5885e-01, -5.7916e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 4.0005e-01,  1.3123e+00,  2.2420e+00,  ..., -2.9605e-01,\n",
       "            9.7862e-01, -1.3111e+00],\n",
       "          [ 8.6978e-02, -2.5881e+00,  1.8382e+00,  ..., -1.7631e-01,\n",
       "            1.6183e+00, -8.1561e-01],\n",
       "          [-5.8209e-01, -1.0193e+00,  1.3990e+00,  ..., -2.1031e+00,\n",
       "            2.0747e+00, -1.3559e+00],\n",
       "          ...,\n",
       "          [-6.4262e-01, -2.5701e+00,  3.1735e-01,  ...,  6.6179e-01,\n",
       "            3.0135e+00, -1.3257e+00],\n",
       "          [ 1.0556e+00, -7.8105e-01,  4.2701e-01,  ..., -4.6140e-01,\n",
       "           -2.7540e-01,  2.0618e-01],\n",
       "          [-2.0138e-01,  1.2802e-02, -6.5603e-02,  ...,  6.8734e-01,\n",
       "           -1.2535e-01, -8.7792e-03]],\n",
       "\n",
       "         [[ 5.4935e-01,  1.1324e-02,  6.1974e-01,  ...,  7.7895e-01,\n",
       "            1.1503e+00, -2.0277e-01],\n",
       "          [-3.8099e-02, -3.0060e+00, -1.3606e+00,  ...,  2.3562e+00,\n",
       "           -1.8905e+00, -6.2019e-01],\n",
       "          [-9.2368e-02, -7.3721e-01, -1.5882e+00,  ...,  9.8470e-01,\n",
       "           -8.3680e-01,  4.0687e-02],\n",
       "          ...,\n",
       "          [ 6.0510e-03, -2.0808e+00, -1.3953e+00,  ...,  5.8404e-01,\n",
       "           -1.4337e+00, -1.1009e+00],\n",
       "          [ 2.7430e-02, -6.8043e-01, -1.5112e+00,  ...,  7.0893e-01,\n",
       "            9.4997e-01, -4.4481e-01],\n",
       "          [-2.7404e-01,  3.6537e-01,  3.1712e-01,  ..., -3.8554e-01,\n",
       "            1.0168e-01, -1.9426e-01]],\n",
       "\n",
       "         [[-1.5254e-01, -6.1968e-01, -1.6444e-01,  ...,  2.2128e+00,\n",
       "           -2.6034e-01,  2.3622e+00],\n",
       "          [-6.1994e-01,  6.3543e-01,  5.3299e-02,  ...,  1.0012e-02,\n",
       "           -9.5666e-01,  1.0998e+00],\n",
       "          [-3.9657e-01, -1.8777e-01, -1.3358e+00,  ..., -2.3441e-01,\n",
       "           -1.0987e+00,  1.2346e+00],\n",
       "          ...,\n",
       "          [ 8.4032e-01,  5.4673e-01, -1.6067e+00,  ..., -5.8429e-01,\n",
       "           -4.4383e-01,  1.9931e-01],\n",
       "          [-7.7786e-01, -9.3089e-01, -8.7464e-01,  ..., -1.1631e-01,\n",
       "            1.0616e+00,  1.2049e+00],\n",
       "          [ 3.0651e-01, -3.6946e-01, -4.3212e-01,  ..., -6.8516e-01,\n",
       "           -1.7133e-01,  1.0510e+00]]],\n",
       "\n",
       "\n",
       "        [[[ 1.1219e-01,  1.4574e+00, -1.0132e+00,  ...,  1.5854e+00,\n",
       "           -3.7032e-01,  4.1638e-02],\n",
       "          [-9.6764e-01,  1.6324e-01, -5.4649e-02,  ...,  1.6622e-01,\n",
       "            1.5184e-01, -2.3852e-01],\n",
       "          [ 2.2076e-02,  2.6627e-01,  1.3308e+00,  ..., -2.4989e-01,\n",
       "            4.8156e-01, -4.5815e-01],\n",
       "          ...,\n",
       "          [ 1.3241e+00,  5.3881e-01,  1.3275e+00,  ...,  6.6081e-01,\n",
       "           -7.5624e-02,  8.5058e-01],\n",
       "          [-8.2843e-01, -5.9101e-01,  5.2919e-03,  ..., -1.7666e-02,\n",
       "            2.7185e-01,  4.8871e-01],\n",
       "          [ 6.4817e-01, -4.1085e-01, -2.8325e-01,  ..., -2.5060e-01,\n",
       "           -2.3930e-01,  3.9054e-01]],\n",
       "\n",
       "         [[ 1.8091e+00,  8.6748e-01,  4.8613e-01,  ...,  1.0589e+00,\n",
       "            2.2271e+00, -1.4501e+00],\n",
       "          [-1.2290e+00,  1.5726e+00, -6.1533e-01,  ...,  5.2125e-01,\n",
       "            7.6094e-01, -6.6736e-01],\n",
       "          [ 1.3631e-01,  8.5291e-01,  7.1596e-01,  ..., -4.4893e-01,\n",
       "            7.6178e-01,  4.1773e-01],\n",
       "          ...,\n",
       "          [ 1.9342e+00, -1.5142e-01, -4.7210e-01,  ...,  1.1381e+00,\n",
       "           -6.6733e-01, -6.4768e-01],\n",
       "          [-2.3216e-01,  1.3587e-01, -7.9377e-01,  ...,  8.5912e-01,\n",
       "            6.3273e-01, -5.0209e-01],\n",
       "          [-1.6370e-01,  9.4218e-02, -6.2832e-01,  ...,  5.9162e-01,\n",
       "           -2.0532e-01, -3.2687e-01]],\n",
       "\n",
       "         [[ 4.5323e-01, -3.1760e-03,  1.6973e+00,  ...,  2.2557e-01,\n",
       "           -4.3484e-01,  1.6445e+00],\n",
       "          [-6.6909e-01,  6.8001e-01,  1.0747e+00,  ..., -7.3127e-01,\n",
       "           -2.1466e-01,  7.3720e-01],\n",
       "          [-9.3212e-01,  6.5263e-02,  9.1001e-01,  ..., -9.3988e-01,\n",
       "           -2.0412e+00,  8.2995e-01],\n",
       "          ...,\n",
       "          [-4.0396e-01, -1.4684e+00, -6.4342e-01,  ..., -8.6831e-01,\n",
       "            2.0835e-01,  6.8196e-01],\n",
       "          [ 1.2790e+00, -8.3257e-02, -3.6351e-01,  ...,  5.4402e-02,\n",
       "            1.1123e-01,  3.7416e-01],\n",
       "          [-3.3312e-01,  3.9379e-01,  1.5927e-02,  ..., -3.9232e-01,\n",
       "           -5.1672e-01, -5.8240e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.5154e+00, -1.0752e+00,  1.6806e+00,  ..., -1.0617e+00,\n",
       "            5.2444e-01, -1.0807e+00],\n",
       "          [-1.3719e+00, -1.2482e+00,  1.9628e-01,  ..., -1.2730e+00,\n",
       "            1.3656e+00, -8.9384e-01],\n",
       "          [-5.1202e-01,  1.2936e+00,  6.1100e-02,  ..., -1.4576e+00,\n",
       "           -9.2203e-01,  2.6418e-01],\n",
       "          ...,\n",
       "          [-1.3388e+00, -4.8596e-01,  8.2063e-01,  ..., -1.0399e+00,\n",
       "            2.7299e-01, -6.1471e-01],\n",
       "          [ 7.3159e-01,  4.5834e-01,  2.9985e-01,  ..., -2.1020e+00,\n",
       "            1.8000e-01, -7.2563e-01],\n",
       "          [-1.8703e-01,  4.4194e-04, -8.0599e-02,  ...,  6.4704e-01,\n",
       "           -1.6223e-01, -8.9883e-02]],\n",
       "\n",
       "         [[ 1.6340e+00, -3.2900e+00, -4.2928e-01,  ...,  8.2096e-01,\n",
       "           -4.8467e-01,  3.6938e-01],\n",
       "          [ 2.7932e-01, -1.0487e+00, -1.2778e+00,  ...,  1.0209e+00,\n",
       "           -5.8236e-01, -3.9680e-01],\n",
       "          [ 1.2148e+00, -1.3930e-01, -1.9388e+00,  ...,  1.8474e+00,\n",
       "           -6.1629e-01,  6.4250e-01],\n",
       "          ...,\n",
       "          [ 8.5668e-01, -4.7711e-02, -5.2167e-01,  ...,  2.4136e+00,\n",
       "           -6.6555e-01,  3.1166e-01],\n",
       "          [-8.4470e-01,  5.3903e-01, -1.1561e-01,  ...,  3.7713e-01,\n",
       "            1.4539e+00,  4.3004e-01],\n",
       "          [-2.7515e-01,  3.9761e-01,  3.3924e-01,  ..., -3.5357e-01,\n",
       "            1.0464e-01, -1.8733e-01]],\n",
       "\n",
       "         [[-5.0762e-01, -2.6343e-02, -4.4189e-01,  ..., -7.4984e-01,\n",
       "           -5.9310e-01,  8.1231e-01],\n",
       "          [-1.1684e+00,  5.6902e-01, -1.8845e+00,  ..., -5.6826e-01,\n",
       "           -2.8408e-01,  3.9582e-01],\n",
       "          [-1.7231e+00, -6.6327e-01, -8.0103e-01,  ..., -2.3366e-01,\n",
       "           -4.3137e-02, -7.7468e-01],\n",
       "          ...,\n",
       "          [ 1.1179e+00, -1.6735e+00,  9.0274e-01,  ...,  8.4841e-01,\n",
       "            1.5874e+00,  1.1140e+00],\n",
       "          [ 5.8572e-01, -9.4290e-01, -8.6678e-01,  ..., -1.2437e-03,\n",
       "            1.6401e+00,  1.4166e+00],\n",
       "          [ 3.3068e-01, -3.4781e-01, -4.4871e-01,  ..., -6.4361e-01,\n",
       "           -1.8119e-01,  1.0800e+00]]],\n",
       "\n",
       "\n",
       "        [[[-1.1488e-01, -3.0083e-01,  1.0130e+00,  ...,  1.4796e-02,\n",
       "           -9.5406e-01, -9.3457e-01],\n",
       "          [-2.7520e-01, -1.9521e-01, -2.6841e-01,  ..., -4.2712e-01,\n",
       "            8.5576e-01,  7.6490e-01],\n",
       "          [-1.4911e+00, -1.0503e+00, -2.9880e-01,  ...,  4.9418e-01,\n",
       "           -1.3976e+00, -4.6372e-01],\n",
       "          ...,\n",
       "          [-1.7495e+00, -8.4283e-01, -1.2968e+00,  ..., -9.7307e-02,\n",
       "           -1.8464e+00,  7.5360e-01],\n",
       "          [-4.9807e-01, -6.2592e-01,  3.1803e-01,  ...,  7.5093e-01,\n",
       "           -1.3310e+00, -8.5776e-01],\n",
       "          [ 6.4554e-01, -4.1325e-01, -2.9036e-01,  ..., -2.5747e-01,\n",
       "           -2.4851e-01,  4.1632e-01]],\n",
       "\n",
       "         [[-7.6011e-01,  1.3310e+00,  3.4012e-01,  ...,  1.1865e+00,\n",
       "            1.0706e+00,  8.6858e-01],\n",
       "          [ 7.3691e-01,  9.7557e-01,  7.4044e-02,  ...,  2.6262e-01,\n",
       "            4.4234e-01, -6.2937e-01],\n",
       "          [-8.0572e-01,  1.4930e+00,  1.0228e+00,  ..., -4.9540e-01,\n",
       "            7.0378e-01, -4.4207e-01],\n",
       "          ...,\n",
       "          [ 1.2052e+00,  9.0892e-01,  5.4714e-01,  ...,  7.6001e-01,\n",
       "           -7.8327e-01,  2.9445e-01],\n",
       "          [ 4.3600e-01, -5.1500e-03, -6.9875e-03,  ...,  1.2337e+00,\n",
       "            8.6263e-01, -8.4073e-01],\n",
       "          [-1.6566e-01,  8.4079e-02, -5.8647e-01,  ...,  6.1747e-01,\n",
       "           -2.1210e-01, -3.2400e-01]],\n",
       "\n",
       "         [[ 1.6184e+00, -8.8591e-02,  2.2602e+00,  ..., -3.9295e-01,\n",
       "            2.6890e-01,  4.9873e-01],\n",
       "          [ 7.0831e-01,  7.4226e-01,  1.5014e+00,  ..., -1.9264e+00,\n",
       "           -1.5299e+00, -3.1013e-01],\n",
       "          [-4.0412e-01,  7.5394e-01,  2.3576e+00,  ...,  1.2996e-01,\n",
       "           -5.1098e-01,  1.1975e+00],\n",
       "          ...,\n",
       "          [-4.3835e-01,  1.6208e+00,  7.6803e-01,  ...,  1.9874e-01,\n",
       "            1.7038e-01,  1.6278e+00],\n",
       "          [ 9.7515e-01,  1.6500e-01, -6.2282e-01,  ..., -1.4537e-01,\n",
       "           -1.0118e+00, -6.1777e-01],\n",
       "          [-3.4975e-01,  4.1943e-01,  1.0351e-02,  ..., -3.5830e-01,\n",
       "           -5.3667e-01, -5.7378e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-8.9622e-01,  1.1820e-01, -4.6340e-01,  ..., -9.2394e-01,\n",
       "            1.5277e+00, -1.7817e-01],\n",
       "          [-1.1463e-01,  3.1522e-01, -1.8280e+00,  ..., -2.0989e-01,\n",
       "            5.3054e-01,  1.5620e+00],\n",
       "          [ 1.0178e+00,  7.1570e-01,  3.7934e-01,  ..., -3.0881e-01,\n",
       "            1.6655e+00, -8.4419e-01],\n",
       "          ...,\n",
       "          [ 9.7959e-01, -7.1319e-01,  9.7493e-01,  ..., -8.8502e-01,\n",
       "            1.4453e+00, -5.7209e-01],\n",
       "          [ 2.9553e-01,  2.2953e-01,  7.5058e-01,  ...,  4.4354e-01,\n",
       "           -2.3126e-01,  1.8207e-01],\n",
       "          [-1.8982e-01, -3.2603e-02, -6.7347e-02,  ...,  6.7841e-01,\n",
       "           -1.2971e-01, -2.0525e-02]],\n",
       "\n",
       "         [[ 1.1300e+00, -5.2756e-01, -5.6875e-01,  ...,  1.7768e+00,\n",
       "           -1.0141e+00, -3.2843e-01],\n",
       "          [ 4.1409e-01,  1.3515e+00,  1.0554e+00,  ...,  2.0265e+00,\n",
       "           -3.3179e-01,  2.0586e-01],\n",
       "          [-3.3144e-01, -4.5446e-02,  3.4924e+00,  ...,  3.7070e-01,\n",
       "            8.7234e-01, -9.3210e-01],\n",
       "          ...,\n",
       "          [-4.2072e-01,  3.2418e-01, -3.7437e-02,  ...,  1.7940e+00,\n",
       "           -1.5087e-01,  4.3013e-01],\n",
       "          [-1.1702e+00,  1.9941e-03, -4.5051e-01,  ...,  4.8660e-01,\n",
       "            8.9006e-01, -9.5897e-01],\n",
       "          [-2.7651e-01,  3.6599e-01,  3.5072e-01,  ..., -3.5555e-01,\n",
       "            9.7803e-02, -1.9788e-01]],\n",
       "\n",
       "         [[ 6.6483e-02, -1.1074e+00,  5.4643e-01,  ..., -4.2013e-01,\n",
       "            9.8097e-01,  4.8824e-02],\n",
       "          [ 4.2646e-01,  7.4605e-01, -1.9320e+00,  ...,  5.3151e-01,\n",
       "           -1.6711e-01,  1.4353e+00],\n",
       "          [-8.1857e-01,  1.5858e+00,  4.0270e-01,  ...,  1.1494e+00,\n",
       "            7.1032e-02,  2.4036e+00],\n",
       "          ...,\n",
       "          [-1.5754e+00, -9.6378e-01, -1.4820e+00,  ...,  1.1208e+00,\n",
       "           -5.5048e-01,  1.6157e+00],\n",
       "          [-2.7844e-01, -6.1983e-01, -1.2216e+00,  ...,  2.9416e-01,\n",
       "            4.6930e-01,  1.6687e+00],\n",
       "          [ 3.3836e-01, -3.4733e-01, -4.2046e-01,  ..., -6.8432e-01,\n",
       "           -1.6739e-01,  1.0886e+00]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-5.8512e-01, -3.3137e-01, -2.8705e-01,  ...,  4.2616e-02,\n",
       "            3.3931e-01, -6.2744e-01],\n",
       "          [-2.8276e-01,  1.3595e-01, -2.7298e-01,  ..., -1.0200e-01,\n",
       "            3.1541e-01, -7.5565e-02],\n",
       "          [-3.3066e-01,  2.6551e-01, -5.3121e-02,  ...,  1.1163e-01,\n",
       "            7.2331e-02,  7.0141e-02],\n",
       "          ...,\n",
       "          [-4.1903e-01,  3.4691e-01, -4.0629e-01,  ...,  2.8106e-01,\n",
       "            7.2545e-01, -1.4128e-01],\n",
       "          [-3.9062e-01,  7.6203e-02, -3.2113e-01,  ...,  9.8765e-02,\n",
       "            3.6406e-01, -1.5814e-01],\n",
       "          [ 2.9329e-01, -1.4788e-01,  5.1209e-03,  ..., -2.0207e-01,\n",
       "            5.9157e-01,  7.7932e-02]],\n",
       "\n",
       "         [[-6.3176e-01, -5.3900e-01,  2.9484e-01,  ...,  7.1637e-01,\n",
       "            8.7435e-01, -5.4122e-01],\n",
       "          [-7.1751e-01, -2.2628e-01,  9.3635e-02,  ...,  5.2159e-01,\n",
       "            5.6744e-01, -8.0991e-02],\n",
       "          [-5.8968e-01, -3.7293e-01, -8.9548e-02,  ...,  4.3999e-01,\n",
       "            1.2761e-01,  7.7660e-02],\n",
       "          ...,\n",
       "          [-2.4483e-01, -6.0088e-02, -1.9739e-01,  ...,  4.5945e-01,\n",
       "            4.8130e-01,  2.0189e-02],\n",
       "          [-4.4363e-01, -4.0539e-01,  1.3269e-01,  ...,  5.9112e-01,\n",
       "            5.6889e-01, -9.6816e-02],\n",
       "          [-1.0081e+00, -3.3903e-02, -7.1046e-02,  ...,  5.2756e-01,\n",
       "            4.9795e-01, -1.4237e-01]],\n",
       "\n",
       "         [[ 4.4708e-01, -2.7088e-01, -1.3938e-02,  ...,  9.7573e-01,\n",
       "           -5.2318e-03,  1.9829e-02],\n",
       "          [ 6.0069e-01, -6.7439e-02,  4.0215e-01,  ...,  9.6812e-01,\n",
       "           -1.3701e-01,  1.8085e-01],\n",
       "          [ 5.2282e-01, -1.8029e-01,  2.9882e-01,  ...,  1.1483e+00,\n",
       "           -1.8065e-01,  2.0049e-01],\n",
       "          ...,\n",
       "          [ 3.1685e-01, -3.2062e-01,  3.3317e-01,  ...,  4.4549e-01,\n",
       "            1.8796e-04,  2.4455e-01],\n",
       "          [ 5.6974e-01, -2.1168e-01, -6.9924e-02,  ...,  7.7636e-01,\n",
       "           -1.1232e-01,  2.9155e-01],\n",
       "          [-3.2048e-01, -1.0634e+00, -2.8436e-01,  ...,  8.9367e-01,\n",
       "            2.4996e-01,  1.0519e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.0559e-01, -1.0701e-01,  4.0343e-02,  ..., -1.4320e-01,\n",
       "           -6.0244e-01, -1.4424e-01],\n",
       "          [-1.1497e-01, -1.0029e-01, -3.8907e-02,  ..., -3.2405e-01,\n",
       "           -3.6202e-01, -3.1876e-01],\n",
       "          [ 1.1921e-02, -1.1933e-01, -2.4668e-01,  ..., -1.8837e-01,\n",
       "           -5.3850e-01, -4.1000e-01],\n",
       "          ...,\n",
       "          [-1.8272e-01, -1.8897e-01, -1.5328e-01,  ..., -2.6616e-01,\n",
       "           -5.2540e-01, -1.5387e-01],\n",
       "          [-1.8973e-01, -2.0433e-02, -3.7017e-01,  ..., -1.6114e-01,\n",
       "           -5.5869e-01, -3.9145e-01],\n",
       "          [-2.3394e-01,  2.0408e-01, -2.8031e-01,  ...,  3.3945e-01,\n",
       "           -7.1482e-01,  1.3559e-01]],\n",
       "\n",
       "         [[ 2.1110e-01, -7.8632e-01, -5.1258e-01,  ..., -2.8684e-01,\n",
       "           -6.5969e-02,  6.3110e-01],\n",
       "          [ 4.2137e-01, -7.6025e-01, -1.1986e-01,  ...,  1.8320e-01,\n",
       "           -1.2805e-01, -6.0039e-02],\n",
       "          [ 2.0139e-01, -7.2958e-01,  2.1446e-01,  ...,  2.7651e-01,\n",
       "           -2.3150e-01, -2.4830e-01],\n",
       "          ...,\n",
       "          [ 4.2588e-01, -7.7364e-01, -1.7667e-01,  ...,  8.2985e-02,\n",
       "           -2.2594e-01, -3.8317e-01],\n",
       "          [ 3.8216e-01, -5.5382e-01,  3.9527e-02,  ...,  3.6573e-02,\n",
       "           -6.3786e-01, -4.1151e-01],\n",
       "          [ 1.1170e-01, -6.8384e-01, -6.8412e-01,  ..., -3.4627e-03,\n",
       "           -3.8600e-01, -4.3760e-02]],\n",
       "\n",
       "         [[-7.2146e-02,  5.3563e-01,  1.1900e+00,  ..., -7.7503e-01,\n",
       "            1.0492e+00,  2.9954e-01],\n",
       "          [-8.3739e-02, -3.0758e-01,  3.0471e-01,  ...,  5.4372e-02,\n",
       "            3.8405e-01, -1.3569e-01],\n",
       "          [ 1.0858e-01, -1.6182e-01,  3.4895e-01,  ..., -8.6731e-03,\n",
       "            2.0633e-01, -1.9168e-01],\n",
       "          ...,\n",
       "          [ 1.2926e-01,  1.2260e-01, -4.8921e-02,  ..., -6.0043e-02,\n",
       "            4.6574e-01,  1.2628e-01],\n",
       "          [ 1.5055e-01,  8.5070e-02,  3.4747e-01,  ..., -1.6355e-01,\n",
       "            5.1535e-01,  1.9131e-01],\n",
       "          [-3.4267e-01,  3.2701e-01, -6.9905e-02,  ..., -7.4777e-03,\n",
       "            5.4529e-01,  3.4120e-01]]],\n",
       "\n",
       "\n",
       "        [[[-5.2482e-01, -3.8246e-01, -2.8114e-01,  ..., -3.5499e-03,\n",
       "            3.4176e-01, -6.2263e-01],\n",
       "          [-3.8075e-01, -4.2169e-01, -5.5103e-01,  ..., -1.1530e-01,\n",
       "            6.6437e-01, -5.9636e-02],\n",
       "          [-3.6615e-01, -3.2697e-01, -2.6244e-01,  ...,  9.7406e-02,\n",
       "            4.4249e-01, -2.3231e-01],\n",
       "          ...,\n",
       "          [-2.7508e-01, -8.2379e-02, -2.4231e-01,  ..., -3.7249e-02,\n",
       "            6.6473e-01, -3.6437e-01],\n",
       "          [-4.4944e-01, -5.4863e-02, -2.7950e-01,  ..., -1.5524e-01,\n",
       "            3.4997e-01, -1.6362e-01],\n",
       "          [-2.5755e-01, -1.2044e-01, -8.4018e-02,  ..., -2.0798e-01,\n",
       "            3.8782e-01, -2.5420e-01]],\n",
       "\n",
       "         [[-6.2588e-01, -5.1594e-01,  2.6944e-01,  ...,  7.0119e-01,\n",
       "            8.2109e-01, -5.0254e-01],\n",
       "          [-6.4027e-01, -5.1217e-01,  2.2226e-01,  ...,  4.8679e-01,\n",
       "            8.1186e-01, -3.7377e-01],\n",
       "          [-7.3128e-01, -3.1551e-01, -3.8735e-02,  ...,  6.2364e-01,\n",
       "            7.1406e-01, -1.6149e-01],\n",
       "          ...,\n",
       "          [-4.4603e-01, -1.0138e-01, -1.1504e-01,  ...,  4.2140e-01,\n",
       "            5.2319e-01,  2.1395e-01],\n",
       "          [-4.9238e-01, -2.7983e-01, -1.7560e-01,  ...,  4.8347e-01,\n",
       "            2.8479e-01,  1.6027e-01],\n",
       "          [-6.1068e-01, -2.1593e-01,  3.7958e-02,  ...,  4.8573e-01,\n",
       "            4.7364e-01,  6.9774e-02]],\n",
       "\n",
       "         [[ 4.8396e-01, -2.5650e-01,  7.9022e-03,  ...,  9.9929e-01,\n",
       "           -6.0118e-02, -1.2388e-02],\n",
       "          [ 2.8038e-01, -2.4082e-01,  2.1327e-01,  ...,  9.7663e-01,\n",
       "            8.5635e-03,  1.6813e-01],\n",
       "          [ 1.5091e-01, -1.1090e-01,  1.4874e-01,  ...,  8.7915e-01,\n",
       "            2.9026e-03,  3.3032e-01],\n",
       "          ...,\n",
       "          [-3.1774e-02, -1.0953e-01,  4.1390e-01,  ...,  9.6049e-01,\n",
       "           -9.1034e-02,  6.1097e-02],\n",
       "          [ 2.8601e-01, -2.4643e-01,  3.1338e-01,  ...,  6.1779e-01,\n",
       "           -1.4202e-01,  9.9822e-02],\n",
       "          [-9.8256e-02, -6.1190e-01, -6.2227e-02,  ...,  7.2193e-01,\n",
       "           -1.3711e-01,  1.9108e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.8081e-01, -1.0891e-01,  6.3482e-02,  ..., -1.4683e-01,\n",
       "           -6.3639e-01, -1.3582e-01],\n",
       "          [-1.8719e-01, -2.3481e-01,  2.0866e-01,  ..., -2.3576e-01,\n",
       "           -7.4260e-01, -2.6444e-01],\n",
       "          [-1.6921e-01, -3.7355e-01, -2.1125e-01,  ..., -3.9404e-01,\n",
       "           -8.7538e-01, -1.3652e-01],\n",
       "          ...,\n",
       "          [-1.2625e-01, -2.5397e-01,  2.2499e-01,  ..., -3.2809e-01,\n",
       "           -5.9153e-01, -1.9484e-01],\n",
       "          [-4.2270e-01,  9.9693e-03, -1.3011e-02,  ..., -1.7972e-01,\n",
       "           -5.1829e-01, -2.0361e-01],\n",
       "          [-3.2227e-01,  2.5673e-02, -2.8825e-01,  ..., -1.2818e-02,\n",
       "           -6.0000e-01,  1.5862e-01]],\n",
       "\n",
       "         [[ 2.9728e-01, -8.3853e-01, -5.3390e-01,  ..., -3.1163e-01,\n",
       "           -7.9170e-02,  6.3745e-01],\n",
       "          [ 5.9635e-01, -8.6288e-01, -3.5745e-01,  ..., -2.9274e-02,\n",
       "           -3.4169e-01, -3.0390e-01],\n",
       "          [ 2.6936e-01, -5.5341e-01, -2.2123e-01,  ..., -1.0301e-02,\n",
       "           -4.1221e-03, -4.7108e-01],\n",
       "          ...,\n",
       "          [ 4.4611e-01, -9.4457e-01, -4.5207e-01,  ...,  3.0122e-01,\n",
       "           -2.4286e-01, -5.1576e-02],\n",
       "          [ 4.6938e-01, -6.9719e-01, -1.9193e-01,  ...,  7.4867e-02,\n",
       "           -5.9349e-01, -1.1408e-01],\n",
       "          [ 2.0412e-01, -9.2280e-01, -6.4322e-01,  ..., -1.0212e-01,\n",
       "           -5.7009e-01, -1.6378e-01]],\n",
       "\n",
       "         [[-7.3532e-02,  4.6871e-01,  1.1651e+00,  ..., -7.6924e-01,\n",
       "            1.0701e+00,  2.3325e-01],\n",
       "          [ 9.8672e-02,  5.0216e-02,  4.8484e-01,  ...,  7.1197e-02,\n",
       "            3.2329e-01, -7.0526e-02],\n",
       "          [ 1.2301e-01, -1.4533e-01,  3.4737e-01,  ..., -6.1234e-02,\n",
       "            4.0155e-01, -7.2865e-02],\n",
       "          ...,\n",
       "          [-1.2715e-01, -3.1976e-02,  1.4158e-01,  ...,  8.3762e-02,\n",
       "            4.9138e-01, -2.3508e-01],\n",
       "          [ 1.2380e-01,  3.8178e-01,  2.3220e-01,  ..., -1.0121e-01,\n",
       "            8.5791e-01,  1.0346e-01],\n",
       "          [-7.8226e-02,  2.9634e-01,  2.3517e-01,  ..., -2.0653e-01,\n",
       "            6.7553e-01,  1.5376e-01]]],\n",
       "\n",
       "\n",
       "        [[[-5.3093e-01, -3.3438e-01, -2.6896e-01,  ...,  1.3474e-02,\n",
       "            3.5992e-01, -5.9702e-01],\n",
       "          [-3.4022e-01, -4.4988e-01, -1.6952e-01,  ..., -1.4957e-01,\n",
       "            4.1628e-01, -1.1327e-01],\n",
       "          [-5.0900e-01, -2.8647e-01, -1.0360e-01,  ..., -4.1124e-02,\n",
       "            3.2769e-01, -2.7974e-01],\n",
       "          ...,\n",
       "          [-5.4189e-01, -1.2049e-01, -5.5410e-01,  ...,  2.1293e-01,\n",
       "            4.8144e-01, -7.0394e-03],\n",
       "          [-5.0587e-01, -5.9984e-02, -5.2918e-02,  ..., -1.9246e-02,\n",
       "            5.3290e-01,  4.3853e-03],\n",
       "          [-1.3741e-01, -2.0093e-01, -8.7650e-02,  ..., -3.7500e-01,\n",
       "            4.0943e-01, -1.5047e-01]],\n",
       "\n",
       "         [[-6.0906e-01, -5.2219e-01,  2.5937e-01,  ...,  7.0354e-01,\n",
       "            8.4166e-01, -5.1903e-01],\n",
       "          [-3.3301e-01, -5.1898e-01,  1.0511e-01,  ...,  6.8485e-01,\n",
       "            3.1404e-01, -3.3580e-01],\n",
       "          [-4.7311e-01, -3.6151e-01, -8.7212e-02,  ...,  6.2840e-01,\n",
       "            3.2634e-01, -3.6687e-02],\n",
       "          ...,\n",
       "          [-2.9854e-01, -4.0703e-01,  1.9406e-01,  ...,  6.6666e-01,\n",
       "            3.1098e-01, -3.7866e-01],\n",
       "          [-3.5059e-01, -3.6607e-01,  3.4085e-02,  ...,  6.8685e-01,\n",
       "            4.0999e-01,  5.9743e-02],\n",
       "          [-6.9275e-01, -1.6225e-01, -6.9579e-02,  ...,  4.6568e-01,\n",
       "            5.6143e-01,  5.1522e-02]],\n",
       "\n",
       "         [[ 4.7460e-01, -2.9003e-01,  2.3106e-02,  ...,  1.0213e+00,\n",
       "           -3.4279e-02, -1.6124e-02],\n",
       "          [ 5.4085e-01, -2.6515e-01,  1.6321e-01,  ...,  7.9580e-01,\n",
       "           -2.7468e-01,  1.3309e-01],\n",
       "          [ 2.7111e-01, -4.0515e-01, -4.8361e-03,  ...,  7.4157e-01,\n",
       "           -2.9730e-01,  2.5257e-02],\n",
       "          ...,\n",
       "          [ 1.1141e-01, -1.9421e-01,  9.1011e-02,  ...,  8.1208e-01,\n",
       "           -1.3663e-01,  1.3848e-01],\n",
       "          [ 5.4442e-01, -1.3860e-01,  1.7572e-02,  ...,  7.0055e-01,\n",
       "           -4.2502e-02, -1.2098e-01],\n",
       "          [-1.9688e-01, -8.7265e-01, -1.6334e-01,  ...,  7.5640e-01,\n",
       "            1.6993e-01,  4.0918e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.7003e-01, -1.0435e-01,  6.8701e-02,  ..., -1.2093e-01,\n",
       "           -6.4935e-01, -1.6177e-01],\n",
       "          [-6.2491e-02, -2.4478e-01, -8.0987e-04,  ..., -2.3812e-01,\n",
       "           -6.9072e-01, -4.3097e-01],\n",
       "          [-4.4838e-01, -1.4131e-01,  7.8560e-02,  ..., -8.9444e-02,\n",
       "           -7.6664e-01, -4.6441e-01],\n",
       "          ...,\n",
       "          [-2.6137e-01, -1.8680e-01, -1.2541e-01,  ..., -3.1318e-02,\n",
       "           -5.4669e-01, -3.8117e-01],\n",
       "          [-3.6909e-01, -2.5278e-02, -1.3524e-01,  ...,  1.0494e-02,\n",
       "           -8.1673e-01, -1.5922e-01],\n",
       "          [-1.7891e-01,  1.4132e-01, -2.9756e-01,  ..., -7.4898e-02,\n",
       "           -4.5171e-01,  5.6678e-02]],\n",
       "\n",
       "         [[ 2.6053e-01, -8.1145e-01, -5.1625e-01,  ..., -2.7290e-01,\n",
       "           -9.3351e-02,  6.4493e-01],\n",
       "          [ 2.9961e-01, -7.9138e-01, -4.1664e-01,  ...,  1.8033e-01,\n",
       "           -2.0109e-01,  1.8985e-01],\n",
       "          [ 2.1456e-01, -8.0004e-01, -3.1326e-01,  ..., -1.3530e-01,\n",
       "           -3.4852e-01,  3.1338e-02],\n",
       "          ...,\n",
       "          [ 2.5595e-02, -7.5951e-01, -4.1131e-01,  ...,  1.1164e-01,\n",
       "           -8.0444e-02, -5.9159e-02],\n",
       "          [ 1.7853e-01, -7.6349e-01, -3.7608e-01,  ...,  1.0252e-01,\n",
       "           -4.4784e-01,  2.0751e-01],\n",
       "          [ 1.4861e-01, -8.5091e-01, -8.1604e-01,  ...,  1.0673e-02,\n",
       "           -5.3012e-01, -1.8546e-01]],\n",
       "\n",
       "         [[-7.2134e-02,  4.8706e-01,  1.1757e+00,  ..., -7.6973e-01,\n",
       "            1.0590e+00,  2.4513e-01],\n",
       "          [-4.4844e-02,  2.1981e-01,  4.5618e-01,  ..., -5.6257e-02,\n",
       "            7.7184e-01, -1.8143e-01],\n",
       "          [-7.1259e-02,  1.7914e-01,  5.3799e-01,  ..., -9.7400e-02,\n",
       "            7.3690e-01, -4.7888e-02],\n",
       "          ...,\n",
       "          [-8.4689e-02,  8.5607e-02,  1.9553e-02,  ..., -1.5308e-01,\n",
       "            3.8713e-01,  3.6964e-02],\n",
       "          [ 1.3347e-01,  2.2891e-01,  7.1082e-01,  ..., -2.2681e-01,\n",
       "            5.8420e-01,  1.9219e-01],\n",
       "          [-2.0418e-01, -4.4504e-02,  1.5890e-01,  ..., -7.8310e-02,\n",
       "            6.7164e-01,  5.9903e-02]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 2.4332e-01,  1.0021e-01, -9.9214e-03,  ...,  3.1458e-01,\n",
       "            9.9125e-01, -5.6240e-01],\n",
       "          [ 3.0604e-02,  1.6057e-01,  1.7498e-02,  ..., -2.6721e-01,\n",
       "            9.7527e-01, -7.6956e-01],\n",
       "          [ 1.0643e-01,  1.3643e-01, -5.1580e-02,  ..., -2.9632e-01,\n",
       "            8.9693e-01, -8.9081e-01],\n",
       "          ...,\n",
       "          [ 1.5060e-01,  1.8868e-01,  2.7657e-01,  ..., -3.6604e-01,\n",
       "            6.8707e-01, -6.9262e-01],\n",
       "          [ 6.5090e-02,  2.2416e-01, -2.0970e-02,  ..., -2.3438e-01,\n",
       "            7.5994e-01, -5.8005e-01],\n",
       "          [ 2.7732e-02,  7.0953e-01,  3.3360e-01,  ..., -1.0832e-01,\n",
       "            1.2375e+00, -7.0624e-01]],\n",
       "\n",
       "         [[ 8.4993e-01, -3.6019e-01,  6.6009e-01,  ...,  4.4062e-01,\n",
       "           -1.5865e-02, -1.0181e-01],\n",
       "          [ 8.5494e-01, -2.7861e-01,  5.3871e-01,  ...,  1.2258e-01,\n",
       "           -3.9321e-01,  1.9829e-01],\n",
       "          [ 1.1728e+00, -4.2888e-01,  7.0054e-01,  ...,  3.4808e-01,\n",
       "           -2.0663e-01,  6.0366e-02],\n",
       "          ...,\n",
       "          [ 7.4794e-01, -3.8448e-02,  2.0592e-01,  ...,  2.9603e-01,\n",
       "           -8.9651e-02,  1.6316e-01],\n",
       "          [ 1.0533e+00, -3.8336e-02,  6.6241e-01,  ...,  2.4346e-01,\n",
       "           -2.0155e-01,  1.8359e-01],\n",
       "          [ 6.3210e-01, -2.3534e-01,  2.2677e-01,  ...,  6.7206e-01,\n",
       "            3.8430e-01, -1.6555e-01]],\n",
       "\n",
       "         [[-1.0038e+00,  4.9862e-01, -8.4387e-02,  ...,  1.6559e-01,\n",
       "            3.5159e-01, -6.7928e-02],\n",
       "          [-1.0592e+00,  8.0670e-01,  1.2733e-01,  ..., -1.5916e-01,\n",
       "            1.7773e-01,  7.0548e-02],\n",
       "          [-1.0168e+00,  7.0295e-01,  3.2007e-01,  ..., -1.6446e-01,\n",
       "            3.5106e-01,  9.3508e-02],\n",
       "          ...,\n",
       "          [-1.0983e+00,  4.5036e-01, -1.9625e-01,  ...,  2.2433e-02,\n",
       "            2.6615e-01,  1.1552e-01],\n",
       "          [-1.1779e+00,  4.9587e-01, -2.4199e-01,  ...,  5.8909e-02,\n",
       "            3.8537e-01, -7.4665e-02],\n",
       "          [-1.3760e+00,  9.7595e-01, -3.0949e-02,  ...,  1.0348e-01,\n",
       "            3.1936e-01,  9.1904e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.3397e-01, -4.1592e-01, -3.8189e-01,  ...,  1.9229e-01,\n",
       "            1.9080e-01, -4.4193e-01],\n",
       "          [ 6.3856e-01, -4.7704e-01, -7.5462e-01,  ...,  2.7739e-01,\n",
       "            6.0782e-02,  2.6702e-02],\n",
       "          [ 4.2959e-01, -4.7323e-01, -5.4384e-01,  ...,  1.8648e-01,\n",
       "            4.2367e-02,  9.6638e-02],\n",
       "          ...,\n",
       "          [ 3.5061e-01, -4.5337e-01, -1.0081e+00,  ..., -1.2780e-01,\n",
       "           -1.4747e-01, -1.7054e-01],\n",
       "          [ 2.5881e-01, -3.2990e-01, -6.8991e-01,  ...,  1.1651e-01,\n",
       "            9.3211e-02, -4.1799e-01],\n",
       "          [-1.7372e-01, -3.5458e-01, -9.8354e-01,  ...,  3.4035e-01,\n",
       "            1.8624e-01, -3.4577e-01]],\n",
       "\n",
       "         [[-2.4253e-02,  1.6886e-01, -1.3342e-01,  ...,  6.3440e-02,\n",
       "           -1.9877e+00, -2.3764e-01],\n",
       "          [ 1.8314e-02,  4.3407e-02,  4.5045e-02,  ..., -1.9972e-01,\n",
       "           -1.8948e+00, -1.1946e-02],\n",
       "          [-6.7307e-02, -5.6901e-02,  7.5334e-02,  ..., -4.3961e-02,\n",
       "           -2.0812e+00, -2.7669e-01],\n",
       "          ...,\n",
       "          [-7.6373e-02,  2.7991e-01,  1.0597e-01,  ..., -3.8909e-02,\n",
       "           -2.1154e+00,  2.9471e-01],\n",
       "          [-1.3503e-01,  2.5169e-01, -7.1009e-03,  ...,  2.6390e-02,\n",
       "           -2.2915e+00, -7.4861e-02],\n",
       "          [-1.3044e-01, -1.4130e-01,  3.4644e-01,  ...,  1.3543e-01,\n",
       "           -2.4298e+00, -6.0428e-02]],\n",
       "\n",
       "         [[-4.0543e-01, -7.9246e-01,  5.1626e-01,  ...,  9.6589e-01,\n",
       "           -3.6775e-01,  9.9211e-02],\n",
       "          [-3.7571e-02, -9.5594e-01,  3.2382e-01,  ...,  1.1053e+00,\n",
       "            5.7856e-02, -2.5218e-01],\n",
       "          [ 1.3810e-01, -1.0132e+00,  3.0735e-01,  ...,  1.0873e+00,\n",
       "            1.5201e-01, -1.5522e-01],\n",
       "          ...,\n",
       "          [-5.8210e-02, -9.0614e-01,  3.8855e-01,  ...,  1.3058e+00,\n",
       "           -1.2985e-01, -1.5332e-01],\n",
       "          [-5.4485e-02, -8.8527e-01,  3.7734e-01,  ...,  1.2502e+00,\n",
       "            4.2721e-02, -1.9293e-01],\n",
       "          [ 1.1405e-01, -1.1849e+00,  3.0034e-01,  ...,  9.1598e-01,\n",
       "            3.4201e-01,  2.2732e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 2.5188e-01,  8.3184e-02,  2.1730e-02,  ...,  3.3291e-01,\n",
       "            1.0100e+00, -5.8317e-01],\n",
       "          [-6.1208e-02,  2.9578e-01,  4.1712e-02,  ...,  2.1990e-02,\n",
       "            1.2223e+00, -6.6530e-01],\n",
       "          [ 3.4374e-01,  2.0363e-01,  4.4859e-01,  ..., -2.2949e-02,\n",
       "            9.5118e-01, -5.2719e-01],\n",
       "          ...,\n",
       "          [ 2.9915e-01,  2.8815e-01,  1.7041e-01,  ..., -3.5586e-02,\n",
       "            7.7186e-01, -3.7141e-01],\n",
       "          [ 2.7591e-02,  1.3621e-01, -5.0139e-02,  ..., -9.7545e-02,\n",
       "            9.2587e-01, -8.5249e-01],\n",
       "          [-5.9014e-02,  7.1101e-01,  5.3831e-02,  ..., -1.4174e-01,\n",
       "            1.1008e+00, -7.1147e-01]],\n",
       "\n",
       "         [[ 8.6941e-01, -3.6483e-01,  6.3153e-01,  ...,  4.9314e-01,\n",
       "           -8.3448e-02, -1.1200e-01],\n",
       "          [ 7.3369e-01, -4.8920e-01,  4.7137e-01,  ...,  3.7729e-01,\n",
       "           -1.4379e-01,  3.4880e-01],\n",
       "          [ 1.0145e+00, -5.0945e-01,  5.2700e-01,  ...,  3.5210e-01,\n",
       "            1.3726e-01,  1.7413e-01],\n",
       "          ...,\n",
       "          [ 1.0427e+00, -1.4826e-01,  1.9982e-01,  ...,  3.1403e-01,\n",
       "            4.0908e-02,  1.4120e-01],\n",
       "          [ 9.2442e-01, -3.3799e-01,  6.5792e-01,  ...,  2.0048e-01,\n",
       "            1.3175e-01,  8.1013e-02],\n",
       "          [ 7.8747e-01, -1.5673e-01,  5.2131e-01,  ...,  4.7262e-01,\n",
       "            4.2335e-02, -1.1028e-02]],\n",
       "\n",
       "         [[-9.8242e-01,  4.8998e-01, -8.6484e-02,  ...,  1.4505e-01,\n",
       "            3.4939e-01, -5.8032e-02],\n",
       "          [-9.5638e-01,  5.9012e-01, -1.6538e-01,  ..., -9.7207e-02,\n",
       "            3.5834e-01, -1.4418e-01],\n",
       "          [-1.0200e+00,  6.9211e-01, -1.2609e-01,  ..., -3.1578e-02,\n",
       "            1.7840e-01,  1.7219e-02],\n",
       "          ...,\n",
       "          [-8.4836e-01,  5.1674e-01,  7.7239e-02,  ..., -1.3376e-01,\n",
       "            2.6490e-01, -1.1976e-01],\n",
       "          [-1.2685e+00,  3.0719e-01, -4.4207e-02,  ..., -1.6088e-01,\n",
       "            4.6441e-01, -7.7180e-02],\n",
       "          [-1.2996e+00,  5.5251e-01, -2.2882e-02,  ...,  6.1208e-02,\n",
       "            4.1737e-01,  1.1498e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.4859e-01, -3.8260e-01, -3.6709e-01,  ...,  2.2998e-01,\n",
       "            1.7478e-01, -4.2124e-01],\n",
       "          [ 5.2594e-01, -5.8186e-01, -6.9233e-01,  ...,  2.2759e-01,\n",
       "            1.4777e-01,  1.7336e-02],\n",
       "          [ 3.2754e-01, -6.3958e-01, -5.7301e-01,  ..., -1.7490e-01,\n",
       "            1.0946e-01, -2.1499e-01],\n",
       "          ...,\n",
       "          [ 6.9964e-01, -5.1304e-01, -1.0677e+00,  ...,  1.8353e-01,\n",
       "            2.8732e-01, -4.1381e-04],\n",
       "          [ 3.4620e-01, -4.5700e-01, -6.8293e-01,  ...,  3.5137e-01,\n",
       "            1.2944e-02, -3.5584e-01],\n",
       "          [ 4.3482e-01, -5.8802e-01, -7.0320e-01,  ...,  4.3901e-01,\n",
       "            1.0705e-01, -2.0756e-01]],\n",
       "\n",
       "         [[ 2.3397e-02,  1.5855e-01, -1.1766e-01,  ...,  5.0095e-02,\n",
       "           -2.0081e+00, -2.5700e-01],\n",
       "          [ 3.1544e-01,  3.7130e-01,  9.6613e-02,  ...,  1.0112e-01,\n",
       "           -2.3622e+00, -3.7835e-02],\n",
       "          [-3.0909e-02,  3.7298e-01,  1.0657e-01,  ..., -7.9541e-02,\n",
       "           -2.2578e+00,  7.2471e-02],\n",
       "          ...,\n",
       "          [ 1.6437e-01,  1.6914e-01,  1.5165e-01,  ...,  1.4843e-01,\n",
       "           -2.4476e+00, -8.5374e-02],\n",
       "          [-7.7874e-02,  3.8678e-01,  9.3535e-03,  ...,  1.5577e-01,\n",
       "           -2.3302e+00,  7.6587e-02],\n",
       "          [-5.1451e-02,  2.5461e-01,  2.1508e-01,  ...,  1.7359e-01,\n",
       "           -2.0320e+00,  4.2222e-02]],\n",
       "\n",
       "         [[-3.9270e-01, -8.1233e-01,  5.8302e-01,  ...,  9.6418e-01,\n",
       "           -3.2871e-01,  8.7404e-02],\n",
       "          [-2.6242e-01, -9.9807e-01,  7.3480e-01,  ...,  1.3453e+00,\n",
       "            1.8764e-01, -9.9380e-02],\n",
       "          [-3.8835e-01, -6.4153e-01,  6.1296e-01,  ...,  1.2555e+00,\n",
       "            2.8894e-02,  8.6154e-02],\n",
       "          ...,\n",
       "          [-9.8839e-02, -8.5170e-01,  6.3207e-01,  ...,  1.4603e+00,\n",
       "            3.1847e-01, -5.8607e-02],\n",
       "          [-3.1177e-02, -9.3185e-01,  4.1589e-01,  ...,  9.5269e-01,\n",
       "           -1.7082e-01, -2.8683e-01],\n",
       "          [-1.8816e-01, -1.1028e+00,  2.7014e-01,  ...,  8.7620e-01,\n",
       "            1.2362e-01, -8.8105e-02]]],\n",
       "\n",
       "\n",
       "        [[[ 2.5813e-01,  8.3526e-02,  3.4980e-03,  ...,  3.1671e-01,\n",
       "            1.0044e+00, -5.9634e-01],\n",
       "          [-3.4603e-02,  2.2298e-01,  1.1607e-01,  ..., -1.6625e-01,\n",
       "            7.1938e-01, -5.8624e-01],\n",
       "          [-2.6833e-02,  2.3727e-01,  1.7451e-01,  ..., -1.5671e-02,\n",
       "            7.9728e-01, -7.8767e-01],\n",
       "          ...,\n",
       "          [ 2.2850e-01,  2.1132e-01,  9.4534e-02,  ..., -2.1341e-01,\n",
       "            7.2716e-01, -6.2295e-01],\n",
       "          [-1.3066e-01,  2.0465e-01,  4.4363e-02,  ..., -1.9627e-01,\n",
       "            1.1191e+00, -9.0443e-01],\n",
       "          [ 7.5427e-03,  7.5069e-01,  1.5071e-01,  ...,  4.9057e-02,\n",
       "            1.0181e+00, -4.0218e-01]],\n",
       "\n",
       "         [[ 8.7014e-01, -3.6087e-01,  6.5203e-01,  ...,  4.7380e-01,\n",
       "           -6.0224e-02, -1.0925e-01],\n",
       "          [ 9.3699e-01, -7.4568e-02,  6.6095e-01,  ...,  3.7752e-01,\n",
       "            2.9013e-02,  9.2293e-02],\n",
       "          [ 9.5029e-01, -2.0457e-01,  9.2575e-01,  ...,  3.3405e-01,\n",
       "            7.6686e-02, -5.2550e-03],\n",
       "          ...,\n",
       "          [ 9.5625e-01, -3.6230e-02,  7.0491e-01,  ...,  4.8010e-01,\n",
       "            1.8620e-01, -9.8954e-02],\n",
       "          [ 9.9396e-01, -1.7306e-01,  5.3219e-01,  ...,  2.2955e-01,\n",
       "           -9.2570e-02,  2.6856e-01],\n",
       "          [ 7.0332e-01,  2.5369e-01,  1.8703e-01,  ...,  4.9623e-01,\n",
       "            1.2297e-01, -7.9248e-02]],\n",
       "\n",
       "         [[-9.5870e-01,  4.9286e-01, -8.3062e-02,  ...,  1.2982e-01,\n",
       "            3.6124e-01, -7.8039e-02],\n",
       "          [-8.9787e-01,  7.1253e-01, -1.9492e-01,  ..., -1.2995e-01,\n",
       "            2.1203e-01, -8.1766e-02],\n",
       "          [-1.1923e+00,  7.1144e-01,  1.9516e-02,  ...,  3.5378e-03,\n",
       "            3.0783e-01, -6.8286e-02],\n",
       "          ...,\n",
       "          [-1.1040e+00,  4.2084e-01, -1.0873e-01,  ...,  1.1281e-01,\n",
       "            1.7169e-01, -5.4593e-02],\n",
       "          [-1.1022e+00,  6.1251e-01, -4.4352e-02,  ..., -1.0510e-01,\n",
       "            3.3025e-01, -2.5025e-03],\n",
       "          [-1.3102e+00,  3.8932e-01, -4.9576e-02,  ...,  5.0717e-02,\n",
       "            3.9342e-01,  1.5761e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.3255e-01, -3.8180e-01, -4.0223e-01,  ...,  2.2826e-01,\n",
       "            1.7508e-01, -3.9177e-01],\n",
       "          [ 4.4581e-01, -4.9424e-01, -7.5962e-01,  ...,  2.3381e-01,\n",
       "            1.3856e-01, -8.2971e-02],\n",
       "          [ 4.0280e-01, -5.2440e-01, -7.6056e-01,  ...,  2.9376e-01,\n",
       "            2.2903e-01, -2.2849e-01],\n",
       "          ...,\n",
       "          [ 2.2821e-01, -4.9823e-01, -9.2860e-01,  ...,  1.0316e-01,\n",
       "           -3.5872e-02, -7.7443e-02],\n",
       "          [ 4.2297e-01, -3.8233e-01, -8.5628e-01,  ...,  4.6803e-01,\n",
       "           -1.2452e-01, -4.0931e-01],\n",
       "          [ 1.7964e-01, -2.5552e-01, -8.2649e-01,  ...,  4.3949e-01,\n",
       "            1.3594e-01, -2.8788e-01]],\n",
       "\n",
       "         [[ 4.2910e-03,  1.7148e-01, -1.2152e-01,  ...,  5.8856e-02,\n",
       "           -1.9960e+00, -2.5262e-01],\n",
       "          [-1.3131e-01,  3.8406e-01,  6.3240e-02,  ...,  2.4962e-02,\n",
       "           -2.0725e+00, -1.3101e-01],\n",
       "          [ 4.4372e-02,  2.2223e-01, -3.4779e-02,  ...,  3.5334e-02,\n",
       "           -2.2419e+00, -3.0869e-01],\n",
       "          ...,\n",
       "          [-2.1514e-01,  2.7350e-02, -2.9391e-01,  ..., -1.1297e-01,\n",
       "           -1.8898e+00,  3.1632e-01],\n",
       "          [ 1.3633e-01, -7.9190e-03,  1.3023e-01,  ...,  2.0768e-01,\n",
       "           -2.4721e+00, -2.1172e-01],\n",
       "          [-1.2346e-01,  1.5323e-01,  2.1076e-01,  ...,  2.5549e-01,\n",
       "           -2.0017e+00,  1.9467e-02]],\n",
       "\n",
       "         [[-3.7220e-01, -8.0981e-01,  5.4472e-01,  ...,  9.6091e-01,\n",
       "           -3.2033e-01,  9.2393e-02],\n",
       "          [-2.0058e-01, -6.5339e-01,  4.8745e-01,  ...,  8.7417e-01,\n",
       "           -1.3020e-01, -9.8957e-02],\n",
       "          [-2.4098e-01, -6.9476e-01,  4.6980e-01,  ...,  8.9627e-01,\n",
       "           -8.9310e-02, -9.1796e-02],\n",
       "          ...,\n",
       "          [ 1.3074e-02, -9.3904e-01,  2.1223e-01,  ...,  1.0139e+00,\n",
       "            6.0012e-02, -2.2331e-01],\n",
       "          [-1.3409e-01, -9.9700e-01,  3.4388e-01,  ...,  1.0599e+00,\n",
       "           -1.1074e-01, -2.1312e-02],\n",
       "          [-2.5921e-01, -1.2453e+00,  1.4593e-01,  ...,  8.5071e-01,\n",
       "            1.8138e-01,  2.4653e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 6.3084e-01, -7.2749e-01, -2.3834e+00,  ...,  2.0131e+00,\n",
       "            1.6228e+00,  6.6917e-01],\n",
       "          [-5.7791e-01, -2.0632e+00, -6.1502e-01,  ..., -2.9698e+00,\n",
       "            2.4630e+00, -3.7657e+00],\n",
       "          [-5.1991e-01, -5.8539e-01, -6.3950e-01,  ..., -1.1942e+00,\n",
       "           -1.2219e+00, -6.3768e-01],\n",
       "          ...,\n",
       "          [ 3.3242e+00,  1.0442e+00,  2.0594e-01,  ..., -2.7340e-01,\n",
       "            1.4871e+00, -2.0854e+00],\n",
       "          [-1.4398e+00, -1.4442e+00,  1.3744e+00,  ...,  2.5677e+00,\n",
       "            1.2542e-01, -8.4923e-01],\n",
       "          [ 2.8486e+00,  1.8834e+00,  8.4984e-01,  ...,  9.6374e-01,\n",
       "            1.5556e+00, -1.5809e+00]],\n",
       "\n",
       "         [[ 3.4601e+00,  1.3833e-01, -1.2404e+00,  ...,  2.6707e-01,\n",
       "           -9.6984e-01, -1.5775e+00],\n",
       "          [ 1.8823e+00,  1.7605e+00, -1.8004e+00,  ...,  1.0053e+00,\n",
       "           -2.8306e+00,  2.9373e+00],\n",
       "          [ 2.2888e+00, -1.9473e+00, -1.4463e+00,  ...,  6.3863e-01,\n",
       "           -9.3677e-01,  2.3901e+00],\n",
       "          ...,\n",
       "          [ 4.0079e-02,  2.3932e-01,  1.1654e+00,  ...,  1.8722e+00,\n",
       "            1.1271e+00,  2.6255e+00],\n",
       "          [-9.7430e-01, -1.0697e+00, -3.5756e-01,  ...,  1.2763e+00,\n",
       "           -7.3420e-01,  1.2672e+00],\n",
       "          [-1.4253e+00, -2.2904e+00, -9.0997e-01,  ..., -4.9642e-01,\n",
       "            1.6879e+00,  1.3423e+00]],\n",
       "\n",
       "         [[-7.1014e-02, -2.1747e-01,  2.6721e-01,  ..., -2.2274e-01,\n",
       "           -5.4326e-01,  1.5725e+00],\n",
       "          [-1.8329e+00,  1.4545e+00,  4.1932e-01,  ...,  7.8005e-01,\n",
       "           -1.6362e+00, -8.8455e-01],\n",
       "          [-5.0695e-01,  3.7875e-01,  5.1875e-01,  ...,  1.4367e+00,\n",
       "           -1.2543e+00, -2.3639e-02],\n",
       "          ...,\n",
       "          [ 7.7787e-01, -3.9504e-01, -1.5647e+00,  ...,  2.1315e+00,\n",
       "           -1.4589e+00, -1.7893e+00],\n",
       "          [ 7.9133e-01, -7.3706e-01,  1.9611e-01,  ..., -6.2221e-01,\n",
       "           -2.5349e-03,  2.4042e-01],\n",
       "          [-2.2129e+00,  2.6926e+00, -5.3376e-01,  ..., -6.0534e-01,\n",
       "            1.3526e+00, -6.2445e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.7006e+00,  1.6752e-01,  4.2777e+00,  ...,  1.7631e+00,\n",
       "           -7.0603e-01,  1.4940e+00],\n",
       "          [-2.1394e+00, -2.1535e+00,  2.6899e+00,  ..., -9.7410e-01,\n",
       "           -1.3119e+00, -4.3940e-02],\n",
       "          [-1.0935e+00, -1.6844e+00,  7.9222e-01,  ...,  6.5437e-01,\n",
       "           -1.0766e+00,  3.3067e-01],\n",
       "          ...,\n",
       "          [-8.1901e-01, -1.7798e+00, -1.9758e+00,  ...,  1.1615e-01,\n",
       "           -9.1047e-01, -1.0763e+00],\n",
       "          [-4.6090e-01, -2.9836e-01, -9.0043e-01,  ...,  7.7180e-02,\n",
       "           -8.5194e-01, -1.0450e+00],\n",
       "          [-1.0900e+00, -4.0649e-01, -2.5163e+00,  ...,  6.3024e-01,\n",
       "           -3.1112e+00, -2.1630e+00]],\n",
       "\n",
       "         [[ 1.7342e-01, -8.6805e-01, -3.7111e-02,  ..., -2.0049e-01,\n",
       "           -1.4868e-02, -1.6461e+00],\n",
       "          [ 6.4168e-01,  2.1889e+00,  8.8795e-01,  ...,  1.6033e+00,\n",
       "            2.4712e+00,  1.2969e+00],\n",
       "          [ 6.0414e-01,  1.3240e-01, -1.5414e-01,  ...,  6.6032e-01,\n",
       "            2.8082e+00,  1.0573e+00],\n",
       "          ...,\n",
       "          [ 3.7564e-01,  1.7620e+00, -1.8680e-01,  ...,  7.9948e-01,\n",
       "           -2.2080e-01,  8.1628e-01],\n",
       "          [ 2.2489e-01,  1.4793e-01,  2.5620e-01,  ...,  6.2018e-01,\n",
       "           -7.7027e-01,  1.4298e+00],\n",
       "          [ 5.5861e-01, -9.0368e-01, -1.3192e+00,  ...,  3.7432e-01,\n",
       "           -7.4476e-01, -3.3707e-02]],\n",
       "\n",
       "         [[-1.0057e+00, -4.3838e-01,  5.8441e-01,  ...,  3.3813e-01,\n",
       "           -5.6015e-01, -2.9716e+00],\n",
       "          [ 1.1710e+00,  3.5705e+00,  1.4164e+00,  ..., -1.7769e+00,\n",
       "            8.4341e-01, -9.9850e-01],\n",
       "          [ 1.1228e+00,  2.6550e+00,  3.8856e-01,  ..., -6.4078e-01,\n",
       "           -7.0246e-01,  1.8889e+00],\n",
       "          ...,\n",
       "          [ 1.1196e+00,  2.2764e+00, -1.9420e+00,  ..., -1.1090e+00,\n",
       "            1.8873e+00,  2.1476e+00],\n",
       "          [ 2.3407e-01, -3.5789e-01, -4.0033e-01,  ...,  1.5307e+00,\n",
       "           -1.9249e+00,  1.4652e+00],\n",
       "          [-3.2985e-02, -1.0779e+00, -9.2885e-01,  ...,  9.9131e-01,\n",
       "           -1.1726e+00, -9.4781e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 2.9016e+00,  1.6698e+00, -3.5789e+00,  ...,  6.1123e-01,\n",
       "           -2.1679e+00, -1.5531e+00],\n",
       "          [ 7.1931e-01,  1.3826e+00, -9.8297e-01,  ..., -1.8957e+00,\n",
       "           -2.1695e+00, -1.3950e+00],\n",
       "          [-2.8962e+00, -1.8074e+00,  3.5306e-01,  ..., -4.9759e-02,\n",
       "           -1.6489e+00,  1.6114e-01],\n",
       "          ...,\n",
       "          [-2.9402e+00, -6.5056e-02, -5.2435e-01,  ..., -3.7415e-02,\n",
       "            3.6838e-01,  1.9581e+00],\n",
       "          [-1.7896e+00, -1.1285e+00,  2.5632e-01,  ..., -4.0799e-01,\n",
       "            1.3846e+00,  7.1118e-01],\n",
       "          [ 2.8667e+00,  1.9473e+00,  8.0509e-01,  ...,  8.8146e-01,\n",
       "            1.5647e+00, -1.5689e+00]],\n",
       "\n",
       "         [[-1.9045e+00, -2.0517e+00,  2.0750e+00,  ...,  1.4728e+00,\n",
       "           -1.4319e+00,  1.3035e+00],\n",
       "          [ 7.2978e-01, -2.3348e+00, -1.7240e+00,  ..., -5.3351e-01,\n",
       "            9.1334e-01,  1.2858e+00],\n",
       "          [-4.1815e-01,  9.1970e-01, -1.1454e+00,  ..., -1.2516e+00,\n",
       "           -1.4222e+00, -2.9029e-01],\n",
       "          ...,\n",
       "          [ 8.0135e-01,  1.0878e+00, -1.3916e+00,  ...,  2.4203e+00,\n",
       "            5.5165e-01, -9.2088e-01],\n",
       "          [-2.2329e-01, -1.6192e+00, -2.9094e-01,  ...,  1.2490e+00,\n",
       "            2.4048e-01,  4.2207e-01],\n",
       "          [-1.4054e+00, -2.2896e+00, -9.3462e-01,  ..., -4.7127e-01,\n",
       "            1.6615e+00,  1.3398e+00]],\n",
       "\n",
       "         [[-3.6613e-01,  8.5657e-01,  1.0455e+00,  ...,  3.1623e+00,\n",
       "           -5.8875e-01, -6.1775e-01],\n",
       "          [ 7.8477e-01,  6.9372e-01,  5.5219e-01,  ...,  1.2388e+00,\n",
       "           -1.3808e+00,  1.1689e+00],\n",
       "          [ 4.2528e-01,  1.2346e+00,  8.0993e-01,  ...,  9.1877e-01,\n",
       "           -3.2148e-02,  1.0129e+00],\n",
       "          ...,\n",
       "          [-4.9523e-01,  2.4526e-01, -1.5332e+00,  ...,  1.6762e+00,\n",
       "            7.5550e-01, -1.8292e-01],\n",
       "          [-3.5773e-01, -6.8526e-01, -3.2862e-01,  ...,  4.4692e-01,\n",
       "            5.5200e-01, -2.7327e-01],\n",
       "          [-2.2119e+00,  2.6389e+00, -5.3743e-01,  ..., -5.9304e-01,\n",
       "            1.3226e+00, -6.2110e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-4.7640e-01,  5.6133e-01,  7.2038e-02,  ...,  1.4418e+00,\n",
       "           -9.4006e-01, -1.1136e+00],\n",
       "          [-6.1733e-01, -8.3149e-02, -7.2907e-01,  ...,  7.8626e-01,\n",
       "           -2.1779e+00, -7.5345e-01],\n",
       "          [-6.5012e-01,  1.6730e+00,  1.9141e+00,  ..., -1.9734e+00,\n",
       "           -3.0342e+00, -1.5425e-02],\n",
       "          ...,\n",
       "          [-2.6536e+00, -5.2889e-01, -3.3558e-01,  ..., -2.1372e+00,\n",
       "           -7.7438e-01, -9.8939e-01],\n",
       "          [-1.4979e+00, -1.1155e+00,  7.3162e-02,  ..., -1.0197e+00,\n",
       "            9.1982e-01,  6.8541e-01],\n",
       "          [-1.1371e+00, -4.6301e-01, -2.5073e+00,  ...,  6.0330e-01,\n",
       "           -3.0967e+00, -2.1006e+00]],\n",
       "\n",
       "         [[ 3.9981e+00,  7.4376e-01, -2.4482e+00,  ...,  1.0008e+00,\n",
       "            1.9892e-01, -4.2236e-01],\n",
       "          [-4.1901e-01, -1.5540e+00, -2.3978e+00,  ...,  2.2959e-01,\n",
       "            1.4042e+00,  1.4160e+00],\n",
       "          [ 9.6568e-02, -3.4074e+00, -1.8306e+00,  ..., -1.3465e+00,\n",
       "            3.3516e+00,  1.0844e+00],\n",
       "          ...,\n",
       "          [ 2.8195e+00,  1.7517e+00, -2.9267e-01,  ...,  1.3869e+00,\n",
       "           -2.6302e-01, -2.1346e-01],\n",
       "          [ 7.1869e-01,  2.6299e+00,  1.3726e+00,  ..., -7.2967e-01,\n",
       "           -3.8320e-02,  4.7997e-01],\n",
       "          [ 5.1049e-01, -8.3729e-01, -1.2359e+00,  ...,  3.7946e-01,\n",
       "           -6.8940e-01, -9.8680e-02]],\n",
       "\n",
       "         [[ 3.9894e-01,  6.3642e+00, -6.9407e-01,  ..., -2.4204e+00,\n",
       "            9.9661e-02,  1.6602e+00],\n",
       "          [ 2.9009e-01,  2.8875e+00, -5.9823e-01,  ..., -1.5275e+00,\n",
       "            3.0009e-01,  2.7532e+00],\n",
       "          [-1.5683e+00,  3.3301e+00,  1.1391e+00,  ..., -1.8204e+00,\n",
       "            8.8577e-01,  4.0576e-01],\n",
       "          ...,\n",
       "          [ 1.5765e+00, -1.3096e+00,  1.3070e+00,  ..., -1.0788e+00,\n",
       "           -9.3143e-01, -1.0218e+00],\n",
       "          [ 1.7775e+00, -2.0215e+00,  8.7458e-01,  ...,  6.1453e-02,\n",
       "           -1.5923e+00, -1.4318e+00],\n",
       "          [-5.5764e-03, -1.2341e+00, -8.7135e-01,  ...,  1.0287e+00,\n",
       "           -1.1652e+00, -9.7363e-01]]],\n",
       "\n",
       "\n",
       "        [[[-5.0779e-01, -1.9700e+00, -9.3650e-01,  ..., -1.6591e+00,\n",
       "           -6.9785e-03,  5.3409e-01],\n",
       "          [-1.7374e+00, -6.0154e-01, -1.6257e+00,  ..., -2.1644e-01,\n",
       "           -1.9625e+00,  2.7490e-01],\n",
       "          [-2.3513e-02, -2.0759e+00, -4.6458e-02,  ...,  4.3254e-01,\n",
       "           -1.5749e+00,  9.7553e-01],\n",
       "          ...,\n",
       "          [ 3.0789e+00, -4.4055e-01, -3.4675e-01,  ...,  1.9541e+00,\n",
       "           -1.3315e-01,  1.6432e+00],\n",
       "          [ 1.2252e+00, -8.9801e-01,  2.0582e+00,  ...,  1.9425e+00,\n",
       "            4.2721e-01,  2.9842e-01],\n",
       "          [ 2.8364e+00,  1.8975e+00,  7.9497e-01,  ...,  8.4816e-01,\n",
       "            1.5540e+00, -1.6293e+00]],\n",
       "\n",
       "         [[ 2.3793e+00, -1.1867e+00, -2.3291e+00,  ..., -1.2343e+00,\n",
       "           -9.4572e-01,  2.0180e+00],\n",
       "          [ 4.9382e+00, -3.9990e+00, -2.9962e+00,  ..., -4.7748e-01,\n",
       "            7.9452e-01,  1.3536e+00],\n",
       "          [ 2.7432e+00, -5.3089e+00, -4.1076e+00,  ...,  1.4674e+00,\n",
       "           -3.7301e+00,  5.0269e-01],\n",
       "          ...,\n",
       "          [-5.6767e-01, -7.9682e-01, -1.1765e+00,  ...,  1.2523e+00,\n",
       "           -2.1128e+00,  3.1851e+00],\n",
       "          [-5.7568e-01, -2.2751e-01, -1.1989e+00,  ...,  1.2804e+00,\n",
       "            4.1423e-01,  5.9505e-02],\n",
       "          [-1.3452e+00, -2.3237e+00, -9.3611e-01,  ..., -5.1725e-01,\n",
       "            1.7042e+00,  1.2981e+00]],\n",
       "\n",
       "         [[-2.7586e-01, -4.8123e-01,  3.2199e-01,  ...,  1.6814e-01,\n",
       "            1.1290e-01,  2.7903e+00],\n",
       "          [-2.1264e-01,  1.1419e+00, -3.3734e-02,  ...,  7.6985e-01,\n",
       "           -1.8989e+00,  1.1200e+00],\n",
       "          [ 9.2284e-01,  3.1371e-01, -2.3684e-01,  ..., -4.9059e-02,\n",
       "           -3.1292e+00,  2.8026e+00],\n",
       "          ...,\n",
       "          [-1.7308e-01, -3.6384e-02, -1.8387e+00,  ...,  7.6103e-01,\n",
       "           -9.5415e-01,  6.3054e-01],\n",
       "          [-1.3708e-01,  4.1510e-01, -7.4122e-01,  ..., -1.1329e-02,\n",
       "           -9.3376e-01, -5.3924e-01],\n",
       "          [-2.2280e+00,  2.6860e+00, -5.5135e-01,  ..., -5.7953e-01,\n",
       "            1.2958e+00, -6.0781e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.3632e+00,  8.0885e-01,  2.2097e+00,  ...,  2.2191e+00,\n",
       "           -1.4348e+00, -1.5716e+00],\n",
       "          [-4.6122e-01, -1.5553e-01,  5.8473e-01,  ..., -7.9214e-01,\n",
       "           -1.6718e+00, -1.2011e-01],\n",
       "          [-4.3598e+00, -1.0572e+00,  1.6254e+00,  ..., -1.2104e+00,\n",
       "           -4.2577e-01, -3.0375e+00],\n",
       "          ...,\n",
       "          [-9.0881e-01, -1.5174e+00,  1.6125e-01,  ...,  9.1010e-01,\n",
       "            7.4330e-01, -5.6884e-01],\n",
       "          [-2.1174e+00, -6.3606e-02,  2.0314e-01,  ...,  5.4268e-01,\n",
       "            2.4690e-01, -1.0310e+00],\n",
       "          [-1.1271e+00, -4.1908e-01, -2.4783e+00,  ...,  5.9439e-01,\n",
       "           -3.0645e+00, -2.1743e+00]],\n",
       "\n",
       "         [[ 3.3189e-01, -2.3696e+00, -1.2767e+00,  ...,  9.4995e-01,\n",
       "            1.4115e+00, -8.3423e-01],\n",
       "          [-1.0211e+00, -3.7765e-01, -1.0502e+00,  ...,  2.5183e+00,\n",
       "            2.4783e+00, -2.1242e+00],\n",
       "          [-4.6367e+00,  1.1510e-01, -3.4401e-02,  ...,  2.3036e+00,\n",
       "            1.8269e+00, -2.7603e-01],\n",
       "          ...,\n",
       "          [-1.4147e+00,  5.1347e+00,  2.3131e-01,  ...,  2.3876e+00,\n",
       "            2.1231e-01,  2.9031e+00],\n",
       "          [-8.4375e-01,  1.4815e+00,  3.7815e+00,  ...,  2.5324e+00,\n",
       "           -6.9836e-01, -6.5522e-01],\n",
       "          [ 6.1085e-01, -9.3811e-01, -1.2474e+00,  ...,  3.9870e-01,\n",
       "           -6.7640e-01, -4.5533e-02]],\n",
       "\n",
       "         [[ 2.0308e+00,  4.2023e-01, -8.5180e-01,  ..., -1.7495e+00,\n",
       "            1.2662e+00, -1.1434e+00],\n",
       "          [ 5.2018e-01,  9.3804e-01, -2.6510e+00,  ..., -8.4518e-01,\n",
       "            3.6298e-01, -2.0776e+00],\n",
       "          [ 3.2401e+00, -6.0802e-01,  2.6008e+00,  ...,  1.8967e+00,\n",
       "           -2.9042e+00,  6.5318e-01],\n",
       "          ...,\n",
       "          [ 7.5514e-01, -5.0979e+00,  2.5605e+00,  ...,  1.2162e+00,\n",
       "            3.5059e-01,  1.2298e-01],\n",
       "          [ 6.3788e-01, -1.6338e+00, -6.6689e-01,  ...,  3.6576e-01,\n",
       "           -1.4693e+00,  1.9880e-01],\n",
       "          [ 1.7237e-02, -1.1426e+00, -8.6481e-01,  ...,  9.9881e-01,\n",
       "           -1.1987e+00, -1.0463e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 7.1763e-01,  1.0049e+00, -7.2763e-01,  ..., -7.4673e-01,\n",
       "           -7.7166e-01,  4.2701e-01],\n",
       "          [ 1.5789e-01, -1.5522e+00, -8.5904e-01,  ...,  1.3120e+00,\n",
       "            7.2198e-01, -1.4356e+00],\n",
       "          [ 1.1504e-01,  2.6832e-01, -1.8640e+00,  ...,  9.2979e-01,\n",
       "            5.9434e-02,  6.0282e-01],\n",
       "          ...,\n",
       "          [ 8.1931e-01, -1.2086e+00, -9.5496e-01,  ...,  3.1780e+00,\n",
       "           -1.2275e-01, -7.8228e-01],\n",
       "          [ 1.4537e+00,  6.6788e-03, -1.0300e-01,  ...,  3.7533e-01,\n",
       "            1.0007e+00,  8.0798e-01],\n",
       "          [-2.4610e-01, -1.1668e-01, -5.5653e-02,  ...,  1.9123e-01,\n",
       "           -4.2447e-01,  1.3906e-02]],\n",
       "\n",
       "         [[ 6.7891e-01, -1.7798e-01, -1.9615e-01,  ...,  8.7868e-01,\n",
       "           -1.0025e+00,  8.6017e-01],\n",
       "          [ 1.8130e+00, -1.1167e+00, -1.1521e+00,  ...,  4.6056e-01,\n",
       "           -2.0302e-01,  1.5066e+00],\n",
       "          [ 1.0290e+00, -1.4998e+00,  2.7940e-01,  ...,  6.2441e-01,\n",
       "           -1.0772e+00, -2.7265e-01],\n",
       "          ...,\n",
       "          [ 7.2608e-01,  8.7166e-01, -8.8148e-01,  ...,  8.1345e-01,\n",
       "           -9.8572e-01,  5.9791e-01],\n",
       "          [ 9.1964e-01,  3.4613e-01, -9.5528e-01,  ...,  1.3403e+00,\n",
       "            2.8159e-01, -7.7613e-01],\n",
       "          [ 2.7462e-01,  5.5717e-02, -2.9231e-01,  ...,  1.3530e-01,\n",
       "           -3.7542e-01,  5.7798e-01]],\n",
       "\n",
       "         [[ 2.0970e-01,  3.7225e-01, -5.1822e-01,  ...,  1.1035e+00,\n",
       "           -1.2365e+00,  6.3754e-01],\n",
       "          [-8.8126e-01, -1.8709e+00, -1.9276e+00,  ...,  2.1573e+00,\n",
       "            8.2571e-01,  9.8969e-01],\n",
       "          [-7.2060e-01, -1.2929e+00, -2.1977e+00,  ...,  1.8597e+00,\n",
       "            1.1919e+00,  1.5398e-01],\n",
       "          ...,\n",
       "          [ 8.7214e-01, -2.6850e-01, -1.0009e+00,  ...,  2.0439e+00,\n",
       "           -2.1835e-01,  1.1563e+00],\n",
       "          [ 1.0368e+00, -3.5355e-01,  8.8697e-02,  ...,  1.0325e+00,\n",
       "           -9.7195e-01,  4.1698e-01],\n",
       "          [ 8.3024e-01,  2.0403e-01,  2.2708e-01,  ...,  1.0907e+00,\n",
       "           -1.9314e-01,  5.3550e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.6289e-01,  1.0673e+00,  4.0006e+00,  ...,  1.4527e+00,\n",
       "            8.2457e-01,  6.6635e-01],\n",
       "          [-2.5990e-01,  8.4101e-01,  2.2304e+00,  ..., -4.8827e-01,\n",
       "           -7.1540e-01,  1.5008e+00],\n",
       "          [-2.1023e-01,  1.3874e+00,  1.0206e+00,  ..., -1.0180e+00,\n",
       "            5.4525e-01,  9.6086e-01],\n",
       "          ...,\n",
       "          [ 9.6642e-01,  1.4905e+00,  1.9778e+00,  ..., -6.8334e-02,\n",
       "           -3.5383e-01,  1.8286e+00],\n",
       "          [ 4.5983e-01, -8.3130e-01,  4.1956e-01,  ...,  3.7437e-01,\n",
       "           -2.2093e-02,  5.1074e-01],\n",
       "          [ 3.2465e-02,  8.2739e-01,  9.5887e-02,  ..., -7.8758e-02,\n",
       "           -5.1511e-01,  1.3459e-02]],\n",
       "\n",
       "         [[ 5.6421e-02,  1.1986e+00, -2.5572e-01,  ..., -1.4555e-01,\n",
       "            1.3858e-01,  1.4428e+00],\n",
       "          [-6.2197e-01,  8.2936e-01,  1.2408e+00,  ..., -1.2756e+00,\n",
       "            6.0121e-01, -4.2656e-01],\n",
       "          [ 4.2656e-01,  2.6633e-01,  1.2781e-01,  ...,  8.0647e-01,\n",
       "            9.7401e-01, -4.3837e-02],\n",
       "          ...,\n",
       "          [ 1.3059e+00,  9.6072e-01, -5.6015e-01,  ..., -6.2841e-01,\n",
       "            9.7859e-01,  1.9726e+00],\n",
       "          [ 2.5104e-01,  2.4535e-01,  1.1297e+00,  ...,  7.0096e-01,\n",
       "            1.5128e+00,  5.8267e-02],\n",
       "          [ 3.8529e-01, -4.4337e-01, -8.7606e-02,  ..., -4.5906e-02,\n",
       "            7.5957e-01, -2.2123e-01]],\n",
       "\n",
       "         [[-5.4809e-01, -2.3451e-01, -1.4085e+00,  ...,  7.3093e-01,\n",
       "           -1.6713e+00, -2.3181e+00],\n",
       "          [-1.0493e+00,  1.2541e+00, -6.0223e-01,  ...,  2.1240e+00,\n",
       "           -2.4379e+00, -1.8812e+00],\n",
       "          [-2.2196e-01,  6.0534e-01, -4.2784e-01,  ...,  2.4073e+00,\n",
       "           -1.4058e+00, -1.4823e+00],\n",
       "          ...,\n",
       "          [-1.4445e+00, -5.4379e-01, -1.2948e+00,  ...,  2.6875e+00,\n",
       "           -8.1755e-01, -2.1284e+00],\n",
       "          [-1.6890e-01,  3.8269e-01, -5.9133e-01,  ...,  8.9626e-01,\n",
       "           -3.1210e-01, -6.8235e-01],\n",
       "          [ 1.9464e-01, -5.2943e-03,  2.4381e-01,  ...,  3.2368e-01,\n",
       "            3.2959e-01, -4.6481e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.0413e-01,  1.3443e-01,  4.2602e-01,  ...,  3.5497e+00,\n",
       "            3.0640e-03, -5.6910e-01],\n",
       "          [ 1.2392e-01,  1.0916e+00, -1.6070e+00,  ...,  3.0014e-01,\n",
       "           -9.1074e-01,  9.0785e-01],\n",
       "          [ 1.7665e+00,  1.4958e+00, -3.0574e-01,  ...,  1.2176e+00,\n",
       "           -1.0498e+00, -1.0564e+00],\n",
       "          ...,\n",
       "          [ 1.7678e+00,  1.1132e+00, -9.1452e-01,  ...,  3.0367e+00,\n",
       "           -8.2304e-01,  2.2465e-01],\n",
       "          [ 1.6898e-01,  4.4407e-01, -4.2808e-01,  ...,  3.1064e-01,\n",
       "            7.4687e-01,  7.4784e-01],\n",
       "          [-1.9276e-01, -1.3088e-01, -7.8632e-02,  ...,  1.9851e-01,\n",
       "           -4.2502e-01,  5.7468e-02]],\n",
       "\n",
       "         [[ 2.1820e+00,  1.0279e+00, -1.4290e+00,  ...,  2.1262e+00,\n",
       "           -1.3575e-01,  2.0690e+00],\n",
       "          [ 3.6616e-02, -6.3283e-01,  3.0347e-01,  ...,  1.2418e-01,\n",
       "           -9.7173e-01, -2.4694e-01],\n",
       "          [-1.7604e-01,  1.0970e-01, -3.6692e-01,  ..., -4.5444e-01,\n",
       "           -7.2862e-01, -2.2967e-01],\n",
       "          ...,\n",
       "          [ 2.7677e-01, -6.9203e-01, -3.5323e-01,  ..., -4.1278e-01,\n",
       "           -5.6227e-01,  9.5912e-01],\n",
       "          [ 1.0788e+00,  2.4144e-03, -8.4447e-01,  ...,  7.3775e-01,\n",
       "           -1.4207e-01,  3.3488e-01],\n",
       "          [ 2.7815e-01,  6.0504e-02, -2.5983e-01,  ...,  1.4411e-01,\n",
       "           -3.5158e-01,  5.5432e-01]],\n",
       "\n",
       "         [[ 6.7134e-02, -6.6239e-01, -8.5068e-01,  ...,  2.7943e+00,\n",
       "            6.1466e-01, -2.0042e+00],\n",
       "          [-5.5381e-02, -8.5612e-01, -1.2411e+00,  ...,  1.9323e+00,\n",
       "            6.5736e-01, -9.5662e-01],\n",
       "          [ 1.2557e+00, -1.5367e+00,  1.8016e-02,  ...,  9.3092e-01,\n",
       "            2.4451e-01,  1.4533e+00],\n",
       "          ...,\n",
       "          [ 4.5247e-01, -1.3848e+00, -8.8420e-01,  ...,  1.8398e+00,\n",
       "            3.2884e-01,  1.8327e+00],\n",
       "          [ 3.0788e-01, -8.7531e-01, -5.9342e-01,  ...,  9.3744e-01,\n",
       "           -2.9762e-01,  1.4661e-01],\n",
       "          [ 8.3232e-01,  2.0302e-01,  2.1753e-01,  ...,  1.0786e+00,\n",
       "           -1.6221e-01,  5.7397e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-9.1960e-01,  5.8628e-01,  2.0628e+00,  ...,  1.5841e+00,\n",
       "            5.5193e-01,  6.6568e-01],\n",
       "          [-6.4927e-01,  1.8396e+00,  7.7996e-01,  ...,  6.7798e-01,\n",
       "            5.2870e-02,  9.6338e-01],\n",
       "          [-1.1389e+00,  4.1978e-01,  1.1187e+00,  ...,  8.6193e-04,\n",
       "           -7.0055e-01, -8.2198e-02],\n",
       "          ...,\n",
       "          [ 7.8397e-01,  1.0215e+00,  7.8306e-01,  ..., -1.1073e-01,\n",
       "            4.0173e-01,  9.1646e-01],\n",
       "          [ 5.5721e-01, -2.0615e-01,  1.2647e+00,  ..., -2.5356e-01,\n",
       "            2.6472e-01,  1.4765e+00],\n",
       "          [ 2.4895e-02,  8.1240e-01,  6.9553e-02,  ..., -1.0054e-01,\n",
       "           -4.7298e-01, -8.3216e-03]],\n",
       "\n",
       "         [[ 8.1071e-01, -3.8037e-01,  2.6493e-01,  ...,  2.3395e-01,\n",
       "            1.0034e+00,  1.9775e+00],\n",
       "          [ 1.4844e+00,  6.0365e-01,  2.7770e-01,  ...,  1.0028e+00,\n",
       "            1.8581e-01,  4.3242e-01],\n",
       "          [ 2.0507e+00, -6.3843e-01,  1.4329e+00,  ...,  7.2542e-01,\n",
       "           -7.1064e-01,  8.8718e-01],\n",
       "          ...,\n",
       "          [ 1.7932e+00,  1.4033e+00, -5.2378e-01,  ..., -6.2553e-01,\n",
       "           -8.4257e-02,  4.1410e-01],\n",
       "          [-2.3393e-01,  1.0938e+00,  6.8620e-01,  ...,  6.9495e-01,\n",
       "            2.1380e-01, -8.3336e-02],\n",
       "          [ 3.7115e-01, -4.1785e-01, -9.9819e-02,  ..., -2.9858e-02,\n",
       "            7.2429e-01, -2.2216e-01]],\n",
       "\n",
       "         [[-4.7416e-01, -2.5457e-01, -1.2843e+00,  ...,  3.8119e+00,\n",
       "           -8.0821e-01, -1.0720e+00],\n",
       "          [-5.0870e-02, -3.1375e-02, -3.3165e-01,  ...,  2.5062e+00,\n",
       "           -1.2327e+00, -2.0838e+00],\n",
       "          [ 1.4323e+00,  1.6493e-01, -2.8393e+00,  ...,  1.2115e+00,\n",
       "           -4.0199e-01, -3.2097e-01],\n",
       "          ...,\n",
       "          [ 3.1380e-01, -2.7574e-01,  2.6840e-01,  ...,  6.4924e-01,\n",
       "           -2.9870e-01, -7.5566e-01],\n",
       "          [ 6.3187e-01,  7.8346e-02,  3.3634e-01,  ...,  1.0030e+00,\n",
       "           -1.1831e+00, -7.7983e-01],\n",
       "          [ 1.7368e-01, -1.8089e-02,  2.6744e-01,  ...,  3.3479e-01,\n",
       "            3.3686e-01, -4.7023e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 4.7492e-01,  1.1939e-01, -1.1101e+00,  ...,  9.0397e-01,\n",
       "           -2.4305e+00,  2.3481e-01],\n",
       "          [ 1.6740e-01,  4.1536e-02,  7.4889e-01,  ...,  8.4129e-01,\n",
       "           -1.5828e+00,  1.5912e+00],\n",
       "          [ 1.6785e+00, -8.3389e-01,  2.9174e+00,  ..., -3.0752e-01,\n",
       "            2.6705e-01,  3.9002e-01],\n",
       "          ...,\n",
       "          [ 1.9336e+00, -1.1922e+00,  1.1050e-01,  ..., -1.9053e-02,\n",
       "           -1.5592e+00, -1.1657e+00],\n",
       "          [ 4.4474e-01, -1.0607e-01,  2.3458e-01,  ..., -4.3870e-02,\n",
       "            7.8997e-01,  1.7403e-01],\n",
       "          [-2.3750e-01, -1.5095e-01, -1.9467e-02,  ...,  2.2136e-01,\n",
       "           -3.8435e-01,  5.5991e-02]],\n",
       "\n",
       "         [[ 5.7162e-01, -4.9255e-02, -7.8440e-02,  ...,  1.6394e-01,\n",
       "           -1.4697e-01,  3.8569e-01],\n",
       "          [ 5.1799e-01,  6.6333e-01, -3.2397e-01,  ..., -1.2140e+00,\n",
       "           -6.2045e-01, -9.3171e-02],\n",
       "          [ 1.5515e+00, -1.3324e+00,  1.8731e+00,  ..., -1.4403e+00,\n",
       "           -2.8085e-01,  2.7156e+00],\n",
       "          ...,\n",
       "          [-2.5139e-02,  4.2317e-01,  2.7843e-02,  ..., -9.1813e-01,\n",
       "            4.1586e-01, -1.5492e-01],\n",
       "          [ 8.7160e-01,  5.1273e-01, -2.1159e-01,  ...,  4.1266e-01,\n",
       "            1.0391e-01,  1.4675e-01],\n",
       "          [ 2.6593e-01,  4.5972e-02, -2.9778e-01,  ...,  1.1481e-01,\n",
       "           -3.2527e-01,  5.9287e-01]],\n",
       "\n",
       "         [[-1.4629e+00, -1.5798e+00, -3.8572e-01,  ...,  1.8390e+00,\n",
       "           -1.9691e-01,  5.8485e-01],\n",
       "          [-4.5522e-01, -1.7494e+00, -3.1170e-01,  ..., -3.5522e-01,\n",
       "            1.7823e+00, -5.0703e-01],\n",
       "          [-2.0684e-01, -8.6788e-01, -9.3610e-01,  ...,  1.2784e+00,\n",
       "            9.8572e-01,  1.4270e-01],\n",
       "          ...,\n",
       "          [ 9.7032e-01, -3.4275e-01,  1.5830e+00,  ...,  1.1726e-01,\n",
       "           -1.0858e-02,  2.9187e-01],\n",
       "          [ 1.0601e-01, -2.4316e-01,  1.0631e+00,  ..., -2.8148e-01,\n",
       "           -6.2510e-01,  2.3295e-01],\n",
       "          [ 8.3940e-01,  2.1967e-01,  2.5759e-01,  ...,  1.0569e+00,\n",
       "           -1.7393e-01,  5.2237e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.0721e-01,  8.5335e-01,  1.4743e+00,  ...,  1.7252e+00,\n",
       "            5.0400e-01,  1.0175e+00],\n",
       "          [ 1.9635e-01, -6.1655e-01,  2.6478e+00,  ...,  1.6425e-01,\n",
       "            1.1904e+00,  1.7133e+00],\n",
       "          [ 2.1080e-01,  1.8665e+00,  2.4326e+00,  ...,  1.4003e-01,\n",
       "           -1.4110e+00,  1.9888e+00],\n",
       "          ...,\n",
       "          [ 2.4746e+00,  1.8347e+00, -8.9881e-01,  ...,  6.1365e-01,\n",
       "            3.7409e-01,  1.1227e+00],\n",
       "          [ 1.0325e+00, -4.4587e-01, -1.7161e-01,  ...,  4.5500e-02,\n",
       "           -1.0479e-01,  1.2493e+00],\n",
       "          [ 1.8034e-02,  8.4686e-01,  7.0687e-02,  ..., -8.9248e-02,\n",
       "           -4.8512e-01,  3.7562e-02]],\n",
       "\n",
       "         [[-8.3640e-01, -1.2175e-01, -1.3184e+00,  ...,  1.4508e-02,\n",
       "            2.6210e-01, -1.3282e-01],\n",
       "          [ 1.4895e+00,  9.8182e-01,  8.8559e-01,  ..., -1.2788e-01,\n",
       "            6.4061e-01,  1.4000e+00],\n",
       "          [ 1.5218e+00,  4.3940e-01,  5.4321e-01,  ..., -1.5200e-01,\n",
       "            1.1390e-01, -2.6442e-01],\n",
       "          ...,\n",
       "          [ 1.0288e+00, -1.1687e+00,  1.3080e+00,  ..., -2.5527e-03,\n",
       "            4.7891e-01,  5.8494e-01],\n",
       "          [ 1.4252e+00,  5.5181e-01,  1.5815e+00,  ...,  3.9163e-01,\n",
       "            4.0422e-01,  2.9767e-02],\n",
       "          [ 3.9468e-01, -4.5186e-01, -5.9908e-02,  ..., -5.3821e-02,\n",
       "            7.4291e-01, -2.0193e-01]],\n",
       "\n",
       "         [[ 6.0549e-01, -8.7109e-01, -1.4368e+00,  ...,  1.4461e+00,\n",
       "            4.5958e-01, -2.9319e+00],\n",
       "          [-5.2344e-01, -3.4665e-01, -1.1466e+00,  ...,  1.7779e+00,\n",
       "            9.1757e-01, -3.1472e+00],\n",
       "          [-1.2241e+00, -1.6865e-02, -2.0726e+00,  ...,  3.5158e+00,\n",
       "            1.5468e+00, -1.7833e+00],\n",
       "          ...,\n",
       "          [-4.7817e-01,  2.0744e-01, -1.0969e+00,  ...,  5.6689e-02,\n",
       "            1.3271e-01, -1.9067e+00],\n",
       "          [ 2.3373e-02, -3.6890e-01, -7.1632e-01,  ...,  1.3425e+00,\n",
       "           -5.8070e-01, -5.5166e-01],\n",
       "          [ 1.7604e-01, -2.3918e-02,  2.3474e-01,  ...,  3.1241e-01,\n",
       "            3.5319e-01, -4.6259e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[-1.0605e+00,  2.0616e-02,  4.2173e-01,  ..., -8.6715e-02,\n",
       "           -4.0128e-02, -4.5409e-01],\n",
       "          [-6.9535e-01,  2.3586e-02,  7.7088e-02,  ..., -2.6843e-01,\n",
       "           -3.8261e-01, -8.8708e-01],\n",
       "          [-6.1448e-01,  1.1007e-02, -2.5895e-02,  ..., -2.5943e-01,\n",
       "           -5.1284e-01, -8.4459e-01],\n",
       "          ...,\n",
       "          [-8.6212e-01, -8.4782e-02,  2.3862e-01,  ..., -1.1549e-01,\n",
       "           -3.7789e-01, -5.8733e-01],\n",
       "          [-6.2072e-01, -1.0061e-01,  4.8626e-01,  ..., -6.9952e-02,\n",
       "           -3.8021e-01, -6.0602e-01],\n",
       "          [-1.2796e+00, -2.7304e-02,  2.5388e-01,  ..., -3.5313e-01,\n",
       "           -6.6157e-01, -5.9757e-01]],\n",
       "\n",
       "         [[ 6.2762e-01,  4.9884e-01,  7.1872e-01,  ...,  3.8899e-01,\n",
       "           -4.0507e-01, -1.4725e+00],\n",
       "          [ 2.7052e-01, -7.9133e-02,  8.5353e-02,  ...,  4.3670e-02,\n",
       "           -1.3458e-01, -6.5533e-01],\n",
       "          [ 3.3110e-02, -1.0010e-01,  2.1929e-01,  ..., -1.9832e-02,\n",
       "           -1.7633e-01, -6.6884e-01],\n",
       "          ...,\n",
       "          [ 3.1807e-01, -7.0978e-02,  2.3994e-01,  ..., -2.8590e-01,\n",
       "           -6.0532e-01, -6.7639e-01],\n",
       "          [ 1.4187e-01, -3.5376e-01,  7.6677e-02,  ..., -1.6671e-01,\n",
       "           -2.3064e-01, -7.6839e-01],\n",
       "          [ 6.4428e-01, -8.4835e-03,  4.5646e-01,  ..., -1.3230e-01,\n",
       "           -5.0445e-01, -6.8496e-01]],\n",
       "\n",
       "         [[ 4.7706e-01, -6.2192e-01,  5.3218e-01,  ...,  1.7110e-01,\n",
       "           -4.6626e-01,  8.1183e-01],\n",
       "          [ 4.1615e-01, -8.1094e-02,  3.2448e-01,  ..., -5.8537e-01,\n",
       "            1.4588e-02,  3.5332e-01],\n",
       "          [ 1.8006e-01, -4.7068e-02,  1.5614e-01,  ..., -4.6104e-01,\n",
       "            2.8201e-01,  2.9729e-01],\n",
       "          ...,\n",
       "          [ 6.3369e-02,  1.2669e-01,  2.2139e-01,  ..., -1.4256e-01,\n",
       "           -1.0274e-02,  3.6067e-01],\n",
       "          [-1.5518e-01, -1.5540e-01,  4.7168e-01,  ..., -1.7144e-01,\n",
       "            2.3308e-02,  3.4198e-01],\n",
       "          [ 2.1070e-01, -2.0267e-01, -1.1178e-01,  ..., -2.7799e-01,\n",
       "           -1.0649e-01,  1.9406e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-3.2100e-01, -4.8088e-01,  1.2167e-01,  ..., -1.6798e-01,\n",
       "           -8.6214e-02,  6.4926e-01],\n",
       "          [ 4.7671e-02, -2.5447e-01,  1.5590e-01,  ..., -2.2139e-01,\n",
       "            3.9599e-01,  5.6814e-01],\n",
       "          [-6.8095e-02, -2.9406e-01,  7.4879e-02,  ..., -3.7737e-01,\n",
       "            2.7106e-01,  2.7409e-01],\n",
       "          ...,\n",
       "          [-4.4400e-02,  3.7065e-02,  3.1762e-01,  ..., -1.2337e-01,\n",
       "            3.0494e-01,  2.1261e-01],\n",
       "          [-2.5800e-02, -1.1905e-01,  3.4268e-01,  ..., -1.8048e-01,\n",
       "            3.6480e-01,  3.2802e-01],\n",
       "          [ 1.2057e-01, -1.2133e-01,  5.7505e-02,  ...,  1.6521e-01,\n",
       "            4.7003e-01,  1.4560e-01]],\n",
       "\n",
       "         [[-3.2028e-01,  1.0337e-01, -8.7354e-02,  ..., -1.9086e-03,\n",
       "            5.4742e-01,  9.6083e-02],\n",
       "          [-4.4661e-02,  3.0699e-01, -3.4782e-01,  ...,  1.1427e-01,\n",
       "            6.3900e-01,  2.3502e-01],\n",
       "          [-2.8393e-01,  4.5811e-01, -8.1929e-02,  ..., -4.1925e-02,\n",
       "            6.4281e-01,  1.3937e-01],\n",
       "          ...,\n",
       "          [-1.9340e-01,  2.4943e-01, -4.5547e-01,  ...,  7.3392e-02,\n",
       "            4.3527e-01,  2.4945e-01],\n",
       "          [-1.2139e-01,  2.7857e-01, -2.3184e-01,  ..., -6.2393e-02,\n",
       "            4.0382e-01,  2.2715e-01],\n",
       "          [-3.3212e-01, -3.0357e-03, -3.0653e-01,  ..., -2.5364e-01,\n",
       "            1.2152e-01, -6.8494e-02]],\n",
       "\n",
       "         [[-6.0748e-01,  1.2657e-01, -2.5536e-01,  ..., -4.1805e-01,\n",
       "           -1.3870e-01, -2.5626e-01],\n",
       "          [-5.6913e-01,  1.6619e-01, -1.9692e-01,  ..., -3.3278e-03,\n",
       "            8.5025e-02, -7.3150e-01],\n",
       "          [-4.7435e-01,  1.8253e-01, -6.3813e-02,  ...,  2.4424e-01,\n",
       "            1.9291e-01, -6.0265e-01],\n",
       "          ...,\n",
       "          [-2.9872e-01, -2.7238e-02,  4.2129e-01,  ...,  6.7418e-02,\n",
       "            4.6878e-01, -8.2030e-01],\n",
       "          [-3.8736e-01,  3.8130e-02,  2.3120e-01,  ...,  3.3307e-02,\n",
       "            4.4102e-01, -6.0917e-01],\n",
       "          [-3.6589e-01,  1.2322e-01, -2.1125e-01,  ..., -1.1738e-01,\n",
       "            2.2604e-01, -4.5876e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.0613e+00,  3.6331e-02,  4.4710e-01,  ..., -9.0241e-02,\n",
       "           -4.5710e-02, -4.5369e-01],\n",
       "          [-6.1857e-01, -4.7189e-02,  2.7161e-01,  ..., -1.0124e-01,\n",
       "           -3.2230e-01, -2.3559e-01],\n",
       "          [-6.7658e-01, -7.8200e-02,  4.4616e-01,  ..., -6.5061e-02,\n",
       "           -2.1513e-01, -3.9411e-01],\n",
       "          ...,\n",
       "          [-7.9054e-01, -4.6087e-02,  3.5811e-01,  ..., -1.5030e-01,\n",
       "           -3.3985e-01, -5.7274e-01],\n",
       "          [-7.5276e-01,  3.9731e-02,  2.6272e-01,  ...,  7.6903e-02,\n",
       "           -2.2437e-01, -4.8419e-01],\n",
       "          [-9.4255e-01, -2.0628e-01,  2.2486e-01,  ..., -1.7355e-01,\n",
       "           -4.4307e-01, -5.1221e-01]],\n",
       "\n",
       "         [[ 6.0564e-01,  4.8510e-01,  6.7946e-01,  ...,  3.7796e-01,\n",
       "           -4.0160e-01, -1.4450e+00],\n",
       "          [ 4.6237e-01,  6.8023e-02,  2.9021e-01,  ...,  1.0231e-01,\n",
       "           -4.3532e-01, -7.2888e-01],\n",
       "          [ 4.1439e-01, -7.0465e-02,  2.5996e-01,  ...,  3.8236e-02,\n",
       "           -5.8437e-01, -7.2061e-01],\n",
       "          ...,\n",
       "          [ 2.2504e-01, -1.5679e-01,  3.6767e-01,  ..., -2.5116e-01,\n",
       "           -4.7051e-01, -5.5037e-01],\n",
       "          [ 2.7294e-01,  1.0238e-01,  2.9911e-01,  ...,  1.8116e-02,\n",
       "           -6.1528e-01, -5.1307e-01],\n",
       "          [ 4.9877e-01,  6.5534e-02,  4.1602e-01,  ..., -1.0941e-02,\n",
       "           -4.9828e-01, -8.4363e-01]],\n",
       "\n",
       "         [[ 5.2047e-01, -5.7216e-01,  5.4354e-01,  ...,  1.6024e-01,\n",
       "           -5.1558e-01,  7.8924e-01],\n",
       "          [ 3.3682e-01, -3.4856e-01,  4.2643e-01,  ..., -2.1449e-01,\n",
       "            2.5411e-02,  3.8770e-01],\n",
       "          [ 6.2411e-03, -1.2323e-01,  1.3124e-01,  ..., -2.3514e-01,\n",
       "            4.4308e-01,  4.5103e-01],\n",
       "          ...,\n",
       "          [ 4.3350e-01,  2.1695e-02,  4.1722e-01,  ..., -3.5880e-02,\n",
       "           -2.0534e-01,  4.5049e-01],\n",
       "          [ 7.8606e-02,  7.9486e-03,  4.4999e-01,  ..., -1.0110e-01,\n",
       "            1.6601e-01,  3.4487e-01],\n",
       "          [ 3.2381e-01, -8.9860e-02,  2.2336e-01,  ..., -1.9462e-01,\n",
       "            1.2117e-01,  3.2180e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.9986e-01, -5.0392e-01,  8.7227e-02,  ..., -1.6801e-01,\n",
       "           -5.6724e-02,  6.4789e-01],\n",
       "          [ 5.8723e-02,  1.1591e-01,  3.9798e-02,  ..., -1.9508e-01,\n",
       "            3.3718e-01,  2.3266e-01],\n",
       "          [ 9.3389e-02, -4.5768e-02,  2.0025e-01,  ..., -1.1342e-01,\n",
       "            3.4007e-01,  3.1133e-01],\n",
       "          ...,\n",
       "          [ 4.5949e-02, -1.8426e-01, -1.8034e-02,  ..., -2.9414e-02,\n",
       "            1.7094e-01,  2.6402e-01],\n",
       "          [-9.0471e-02,  7.9273e-02,  1.5009e-01,  ..., -3.9055e-01,\n",
       "            3.5357e-01,  2.7485e-01],\n",
       "          [ 4.3126e-02, -1.1677e-01,  1.4873e-01,  ..., -6.6036e-02,\n",
       "            4.2135e-01,  3.8174e-01]],\n",
       "\n",
       "         [[-2.8310e-01,  1.1588e-01, -1.1955e-01,  ...,  4.2298e-02,\n",
       "            5.3286e-01,  1.0830e-01],\n",
       "          [-3.4888e-01,  2.3839e-01, -3.5979e-01,  ...,  6.9875e-02,\n",
       "            2.5289e-01,  2.6220e-01],\n",
       "          [-2.7849e-01,  1.6761e-01, -2.8802e-01,  ..., -2.8026e-02,\n",
       "            5.1228e-01,  1.0400e-01],\n",
       "          ...,\n",
       "          [-5.1410e-01,  1.6626e-01, -3.2752e-01,  ...,  2.7864e-02,\n",
       "            4.4011e-01,  2.4348e-01],\n",
       "          [ 1.3746e-01,  2.7911e-01, -4.2517e-01,  ...,  1.2238e-01,\n",
       "            2.8276e-01,  1.2350e-01],\n",
       "          [-9.8000e-02,  1.3415e-01, -3.2432e-01,  ..., -2.7523e-02,\n",
       "            2.6121e-01,  1.3965e-02]],\n",
       "\n",
       "         [[-6.2963e-01,  9.2923e-02, -2.8219e-01,  ..., -4.4653e-01,\n",
       "           -1.5675e-01, -2.6090e-01],\n",
       "          [-2.6948e-01,  3.4743e-01, -3.0268e-01,  ..., -2.1784e-01,\n",
       "           -1.3508e-02, -5.0847e-01],\n",
       "          [-4.7426e-01,  1.5899e-01, -5.2532e-03,  ..., -7.1617e-02,\n",
       "            2.7386e-01, -5.3437e-01],\n",
       "          ...,\n",
       "          [-4.2122e-01, -8.8421e-03, -2.0821e-01,  ...,  8.3434e-02,\n",
       "           -1.1714e-03, -4.4792e-01],\n",
       "          [-2.9349e-01,  1.3538e-01,  4.0926e-01,  ..., -7.7523e-03,\n",
       "            2.5593e-01, -6.4393e-01],\n",
       "          [-3.9541e-01, -7.0229e-02, -8.0106e-03,  ..., -1.6036e-01,\n",
       "            2.4087e-01, -7.5043e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.0723e+00,  3.0754e-02,  4.4169e-01,  ..., -1.0106e-01,\n",
       "           -4.3034e-02, -4.4826e-01],\n",
       "          [-1.0046e+00, -3.3563e-02,  4.3473e-01,  ..., -1.3139e-01,\n",
       "           -1.4209e-01, -4.8624e-01],\n",
       "          [-9.4565e-01, -8.1187e-02,  3.7140e-01,  ..., -1.8901e-01,\n",
       "           -3.7527e-01, -3.7424e-01],\n",
       "          ...,\n",
       "          [-7.9302e-01, -9.1364e-02,  3.2664e-01,  ..., -8.3830e-03,\n",
       "           -3.5025e-01, -6.4906e-01],\n",
       "          [-8.4348e-01, -9.8096e-02,  3.2893e-01,  ..., -1.6811e-02,\n",
       "           -5.1188e-01, -4.6828e-01],\n",
       "          [-1.0252e+00, -6.0103e-02,  2.6237e-01,  ..., -2.0350e-01,\n",
       "           -3.9407e-01, -3.3325e-01]],\n",
       "\n",
       "         [[ 6.0121e-01,  4.8287e-01,  6.8865e-01,  ...,  3.8623e-01,\n",
       "           -3.7279e-01, -1.4664e+00],\n",
       "          [ 3.2654e-01, -1.6921e-02,  3.7496e-01,  ..., -1.0032e-02,\n",
       "           -2.5156e-01, -9.6306e-01],\n",
       "          [ 3.4030e-01,  1.6056e-01,  2.3673e-01,  ...,  9.0775e-02,\n",
       "           -4.3397e-01, -8.9368e-01],\n",
       "          ...,\n",
       "          [ 4.4111e-01, -9.2951e-02,  2.4171e-01,  ..., -2.9368e-01,\n",
       "           -2.9163e-01, -8.5791e-01],\n",
       "          [ 2.6721e-01,  6.0244e-02,  3.7252e-01,  ..., -1.0334e-01,\n",
       "           -5.0010e-01, -1.0716e+00],\n",
       "          [ 4.4909e-01, -7.3896e-02,  5.1876e-01,  ..., -2.5051e-01,\n",
       "           -4.5335e-01, -6.6313e-01]],\n",
       "\n",
       "         [[ 4.9966e-01, -5.8606e-01,  5.3965e-01,  ...,  1.4083e-01,\n",
       "           -4.9181e-01,  7.7336e-01],\n",
       "          [ 1.6241e-01, -3.6097e-01,  5.3563e-01,  ..., -1.6661e-01,\n",
       "            6.8037e-02,  4.4961e-01],\n",
       "          [ 4.4614e-02, -3.3782e-01,  5.8875e-01,  ..., -2.0604e-01,\n",
       "            1.2339e-01,  4.0387e-01],\n",
       "          ...,\n",
       "          [ 1.3081e-01, -3.5622e-01,  4.4581e-01,  ..., -3.3426e-01,\n",
       "            1.7402e-01,  5.5028e-01],\n",
       "          [ 3.3060e-01, -2.7319e-01,  4.6872e-01,  ..., -1.1515e-02,\n",
       "            1.2927e-01,  4.1886e-01],\n",
       "          [ 2.0037e-01, -1.1046e-01,  1.7701e-01,  ..., -1.9808e-01,\n",
       "           -1.0459e-01,  2.4137e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.8306e-01, -4.7298e-01,  8.6634e-02,  ..., -1.7903e-01,\n",
       "           -5.4654e-02,  6.6899e-01],\n",
       "          [ 6.2785e-02, -2.8007e-01,  1.9254e-01,  ..., -1.6016e-01,\n",
       "            1.3920e-01,  5.3339e-01],\n",
       "          [ 8.9883e-02, -3.2362e-01,  2.2373e-01,  ...,  1.1415e-02,\n",
       "            9.3626e-02,  4.6896e-01],\n",
       "          ...,\n",
       "          [ 5.1958e-02, -2.1462e-01,  3.7227e-01,  ..., -1.8613e-01,\n",
       "            4.0841e-01,  5.3705e-01],\n",
       "          [-8.2740e-02, -2.8817e-01,  2.8179e-01,  ...,  9.1640e-03,\n",
       "            3.5046e-01,  4.2445e-01],\n",
       "          [-3.2276e-02, -8.9369e-02,  1.0169e-01,  ...,  1.3682e-01,\n",
       "            3.7599e-01,  4.3264e-01]],\n",
       "\n",
       "         [[-2.8636e-01,  1.0938e-01, -1.0385e-01,  ...,  5.1141e-02,\n",
       "            5.4499e-01,  1.1655e-01],\n",
       "          [-3.0063e-01,  1.1836e-01, -3.4038e-01,  ...,  2.3982e-02,\n",
       "            3.8470e-01,  3.2576e-01],\n",
       "          [-3.9713e-01,  1.3052e-01, -9.8102e-02,  ..., -9.0216e-02,\n",
       "            4.0778e-01,  1.2948e-01],\n",
       "          ...,\n",
       "          [-2.1568e-01,  2.1910e-01, -3.9842e-01,  ..., -1.8404e-03,\n",
       "            5.8008e-01,  2.1157e-01],\n",
       "          [-3.9905e-01,  1.3686e-01, -1.5770e-01,  ...,  1.1738e-01,\n",
       "            3.7967e-01,  2.8701e-01],\n",
       "          [-2.9782e-01,  3.6168e-01, -2.5114e-01,  ..., -2.0879e-01,\n",
       "            3.0960e-01,  2.6256e-02]],\n",
       "\n",
       "         [[-6.2470e-01,  1.0430e-01, -2.7552e-01,  ..., -4.4098e-01,\n",
       "           -1.1414e-01, -2.6054e-01],\n",
       "          [-5.1349e-01,  1.4122e-01,  1.3747e-01,  ..., -2.7221e-01,\n",
       "            3.2263e-01, -4.8485e-01],\n",
       "          [-2.6784e-01,  1.5378e-01,  3.5214e-01,  ..., -2.2707e-01,\n",
       "            3.0952e-01, -5.0881e-01],\n",
       "          ...,\n",
       "          [-4.0921e-01,  1.6960e-02,  3.3562e-01,  ..., -2.1096e-01,\n",
       "            4.6905e-01, -8.7396e-01],\n",
       "          [-4.5805e-01,  8.8428e-02, -3.2094e-02,  ..., -1.0551e-01,\n",
       "            2.0562e-01, -5.7112e-01],\n",
       "          [-4.9758e-01,  1.4819e-02, -2.8761e-01,  ..., -3.3288e-01,\n",
       "            3.2298e-01, -5.1378e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 1.4085e-01, -2.3684e-02,  4.0154e-01,  ...,  5.4264e-02,\n",
       "           -7.5099e-01,  7.9200e-04],\n",
       "          [ 1.6209e-01,  2.2954e-01,  4.5493e-01,  ..., -2.0106e-01,\n",
       "           -7.7969e-01,  1.8458e-02],\n",
       "          [ 1.1002e-01,  2.3089e-01,  4.2777e-01,  ..., -1.6087e-01,\n",
       "           -9.1331e-01,  1.6221e-01],\n",
       "          ...,\n",
       "          [-1.3657e-01,  2.0444e-01,  2.7895e-01,  ...,  1.2308e-02,\n",
       "           -6.5307e-01,  3.4266e-01],\n",
       "          [ 6.1241e-02, -8.2518e-02,  3.8677e-02,  ...,  4.1887e-03,\n",
       "           -9.8653e-01,  3.2161e-01],\n",
       "          [ 6.3704e-01,  2.0316e-01,  3.0104e-01,  ...,  9.7930e-02,\n",
       "           -1.1733e+00,  1.7846e-01]],\n",
       "\n",
       "         [[ 2.2480e-01, -1.1277e-01,  5.9243e-02,  ...,  4.6368e-02,\n",
       "            4.6309e-02,  7.3515e-01],\n",
       "          [ 5.8327e-01, -2.6060e-01, -1.0612e-01,  ...,  7.2673e-02,\n",
       "           -2.1529e-02,  7.7430e-01],\n",
       "          [ 4.6703e-01, -1.1800e-01, -6.5540e-02,  ...,  2.0256e-01,\n",
       "           -2.3193e-01,  6.8984e-01],\n",
       "          ...,\n",
       "          [ 3.1361e-01,  1.5838e-01,  1.6710e-01,  ..., -4.5490e-02,\n",
       "           -1.2174e-01,  7.8757e-01],\n",
       "          [ 2.6196e-01,  4.6715e-02,  3.4173e-02,  ..., -6.4588e-02,\n",
       "           -1.6349e-01,  7.3240e-01],\n",
       "          [ 3.5060e-01, -2.5214e-01, -7.2430e-02,  ..., -1.7748e-01,\n",
       "           -7.0252e-01,  5.8273e-01]],\n",
       "\n",
       "         [[ 1.9896e-01,  1.6482e-01, -2.0590e-01,  ...,  2.3588e-01,\n",
       "           -6.0553e-02, -1.0671e-02],\n",
       "          [-7.3521e-02,  2.8232e-01, -9.4057e-02,  ...,  1.0937e-01,\n",
       "           -7.9973e-02,  2.1368e-01],\n",
       "          [ 1.0346e-01,  3.2961e-01, -2.7150e-01,  ..., -1.6077e-02,\n",
       "           -6.6119e-02,  5.9344e-02],\n",
       "          ...,\n",
       "          [ 1.7023e-01, -8.3483e-02,  6.5899e-02,  ...,  8.0147e-02,\n",
       "            8.4112e-02, -3.6977e-01],\n",
       "          [ 3.0991e-01,  4.9746e-02, -9.5673e-02,  ...,  6.7615e-02,\n",
       "            7.3898e-02, -2.0175e-02],\n",
       "          [ 3.4422e-01,  4.2715e-01, -1.1075e-01,  ...,  3.4921e-01,\n",
       "            3.8508e-01,  7.4464e-02]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.3423e-02,  2.5368e-01, -1.0505e-01,  ..., -1.9780e-01,\n",
       "           -2.2417e-01,  1.0537e-01],\n",
       "          [ 1.9467e-02,  3.9280e-01,  2.6562e-01,  ..., -2.7321e-01,\n",
       "           -3.0109e-01, -2.2672e-02],\n",
       "          [ 4.1804e-02,  3.9167e-01, -4.8969e-02,  ..., -1.9763e-01,\n",
       "           -9.1029e-02, -2.0817e-01],\n",
       "          ...,\n",
       "          [ 1.6353e-01,  4.9874e-01,  3.2005e-01,  ...,  7.4219e-03,\n",
       "           -2.3385e-01,  7.6188e-02],\n",
       "          [ 2.6868e-01,  4.5784e-01,  3.3591e-01,  ..., -2.5521e-01,\n",
       "           -2.7635e-01,  2.3735e-01],\n",
       "          [ 1.8768e-01,  6.0566e-01,  2.9336e-02,  ..., -5.2470e-01,\n",
       "           -4.0707e-01,  1.3261e-01]],\n",
       "\n",
       "         [[-3.1403e-01, -3.3988e-01, -1.9267e-01,  ..., -2.5765e-01,\n",
       "            3.2181e-02,  5.7074e-02],\n",
       "          [-2.4329e-02, -8.6694e-02,  1.0305e-01,  ...,  1.2282e-01,\n",
       "            8.3773e-02, -1.0253e-01],\n",
       "          [-7.8454e-02,  8.7747e-02, -1.7367e-01,  ...,  3.0948e-02,\n",
       "            9.3974e-02, -3.1657e-01],\n",
       "          ...,\n",
       "          [-1.5582e-01, -1.7189e-01, -1.7771e-01,  ...,  1.8626e-02,\n",
       "           -6.8327e-02, -8.6154e-03],\n",
       "          [-1.3994e-01, -9.2964e-02,  1.5904e-01,  ..., -1.4304e-01,\n",
       "           -9.1361e-02, -1.4762e-01],\n",
       "          [-2.4220e-01,  1.8056e-01,  1.6999e-01,  ..., -1.2958e-01,\n",
       "            2.4575e-01, -2.8522e-01]],\n",
       "\n",
       "         [[ 1.6844e-02,  9.8760e-02,  1.6054e-01,  ...,  2.0378e-01,\n",
       "            2.1819e-01, -4.1484e-01],\n",
       "          [ 2.6649e-01,  2.0672e-01,  1.9320e-01,  ...,  4.4474e-01,\n",
       "            5.0577e-01, -4.7643e-01],\n",
       "          [ 3.0909e-01,  2.3947e-02,  1.8669e-01,  ...,  2.7857e-01,\n",
       "            6.5941e-01, -6.1912e-01],\n",
       "          ...,\n",
       "          [ 9.8366e-02, -1.3375e-01, -2.4809e-01,  ...,  3.0206e-01,\n",
       "            6.0576e-01, -4.0828e-01],\n",
       "          [-1.2606e-01, -8.2883e-02,  1.0846e-01,  ...,  3.2623e-01,\n",
       "            6.5675e-01, -5.3071e-01],\n",
       "          [ 2.4460e-01, -1.6575e-01,  3.8801e-01,  ...,  4.5548e-01,\n",
       "            4.5549e-01, -3.0001e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.5317e-01, -3.5571e-02,  3.7202e-01,  ...,  5.2405e-02,\n",
       "           -7.3080e-01,  4.0315e-03],\n",
       "          [ 2.1867e-01,  8.3163e-02,  4.0888e-01,  ..., -6.6959e-02,\n",
       "           -8.5713e-01,  4.4943e-01],\n",
       "          [ 1.5012e-01,  7.6156e-03,  1.6738e-01,  ..., -3.1898e-02,\n",
       "           -9.5402e-01,  4.2281e-01],\n",
       "          ...,\n",
       "          [ 3.6408e-03,  1.0145e-01,  2.6748e-01,  ...,  2.4721e-01,\n",
       "           -7.9429e-01,  4.7357e-01],\n",
       "          [-8.3853e-02, -3.7441e-02,  3.5524e-01,  ..., -7.1467e-02,\n",
       "           -8.8839e-01,  1.6107e-01],\n",
       "          [ 7.8983e-02, -1.0107e-01,  3.4506e-01,  ...,  6.7904e-02,\n",
       "           -9.1129e-01, -2.4054e-02]],\n",
       "\n",
       "         [[ 2.5557e-01, -8.9395e-02,  3.6520e-02,  ...,  6.1457e-02,\n",
       "           -1.3239e-02,  7.4089e-01],\n",
       "          [ 5.7876e-01,  4.4548e-02,  7.0716e-02,  ..., -4.0459e-03,\n",
       "           -2.4137e-01,  7.7381e-01],\n",
       "          [ 4.5707e-01, -3.5846e-03, -2.1243e-02,  ..., -1.0433e-01,\n",
       "           -2.7546e-01,  6.7952e-01],\n",
       "          ...,\n",
       "          [ 3.2305e-01, -6.1754e-02,  4.3200e-02,  ...,  9.4049e-02,\n",
       "           -4.0255e-01,  7.6473e-01],\n",
       "          [ 5.1132e-01,  1.2218e-01, -8.2596e-02,  ..., -1.7272e-01,\n",
       "           -1.0109e-01,  6.9337e-01],\n",
       "          [ 6.1547e-01, -4.9883e-02, -1.3895e-01,  ..., -2.4613e-01,\n",
       "           -3.8087e-01,  7.9544e-01]],\n",
       "\n",
       "         [[ 1.6575e-01,  1.8732e-01, -1.7873e-01,  ...,  2.1995e-01,\n",
       "           -2.2238e-02, -6.4991e-02],\n",
       "          [-9.0206e-02,  1.2240e-01, -1.0433e-01,  ...,  1.7204e-01,\n",
       "           -7.7460e-02,  1.3959e-01],\n",
       "          [ 9.0476e-02,  1.9034e-01, -1.4195e-01,  ...,  4.6308e-02,\n",
       "            1.2344e-01, -5.9850e-02],\n",
       "          ...,\n",
       "          [ 2.7567e-01,  1.7908e-01, -1.0208e-01,  ...,  1.4738e-01,\n",
       "            7.7961e-02, -1.1841e-01],\n",
       "          [-3.1543e-03,  1.9366e-01, -6.1829e-02,  ...,  4.7893e-02,\n",
       "           -2.2119e-02, -2.7180e-01],\n",
       "          [ 2.4315e-01,  1.4458e-01, -2.3862e-02,  ...,  2.4678e-01,\n",
       "            3.1557e-01, -1.1576e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.4205e-02,  2.5081e-01, -1.1381e-01,  ..., -2.0697e-01,\n",
       "           -1.9997e-01,  1.3584e-01],\n",
       "          [ 1.8508e-01,  4.5312e-01,  3.3266e-02,  ..., -1.0737e-01,\n",
       "           -1.8212e-01,  3.9630e-02],\n",
       "          [ 2.2802e-01,  4.4200e-01,  1.1288e-01,  ..., -1.6111e-01,\n",
       "           -7.4986e-02,  2.8579e-02],\n",
       "          ...,\n",
       "          [ 1.8304e-01,  3.0056e-01,  7.9171e-02,  ..., -1.3775e-01,\n",
       "           -1.5442e-01,  3.6283e-02],\n",
       "          [ 3.0691e-02,  6.3529e-01,  1.9451e-01,  ..., -2.4818e-01,\n",
       "           -9.0111e-02,  1.5891e-01],\n",
       "          [ 1.4990e-01,  5.5485e-01, -7.2638e-02,  ..., -4.4037e-01,\n",
       "           -3.6812e-01,  1.2437e-01]],\n",
       "\n",
       "         [[-3.0362e-01, -2.9564e-01, -1.5932e-01,  ..., -2.4528e-01,\n",
       "            6.0284e-02,  6.8830e-02],\n",
       "          [-2.0928e-01, -3.1864e-01,  6.9075e-02,  ..., -1.1253e-01,\n",
       "            2.3820e-01,  2.6697e-01],\n",
       "          [-2.5174e-01, -1.7433e-01, -1.3321e-01,  ..., -2.6094e-01,\n",
       "            8.4751e-02, -8.2194e-02],\n",
       "          ...,\n",
       "          [-2.0706e-01, -1.1484e-01, -5.7361e-02,  ...,  4.6149e-02,\n",
       "            3.0465e-01,  1.9993e-01],\n",
       "          [-2.2453e-01, -1.1141e-01, -3.2593e-01,  ...,  6.5626e-02,\n",
       "           -8.4805e-02, -6.4008e-02],\n",
       "          [-2.6566e-01, -8.6189e-02, -1.2823e-02,  ..., -1.0072e-01,\n",
       "            6.5789e-02, -9.0050e-02]],\n",
       "\n",
       "         [[ 8.4074e-03,  8.2548e-02,  1.2868e-01,  ...,  1.2947e-01,\n",
       "            2.5102e-01, -4.3276e-01],\n",
       "          [ 9.1907e-02,  9.4372e-02,  2.5644e-02,  ...,  3.5272e-01,\n",
       "            4.0858e-01, -3.9303e-01],\n",
       "          [ 2.5729e-03, -1.1102e-01,  1.8707e-03,  ...,  3.7127e-01,\n",
       "            5.6416e-01, -5.8116e-01],\n",
       "          ...,\n",
       "          [ 6.9025e-02,  6.5228e-02, -8.2145e-03,  ...,  1.4414e-01,\n",
       "            4.9725e-01, -3.6505e-01],\n",
       "          [-5.6910e-02,  8.0521e-02, -1.1374e-01,  ...,  3.6550e-01,\n",
       "            5.1187e-01, -5.4066e-01],\n",
       "          [-1.1343e-01, -2.1372e-01,  2.1995e-01,  ...,  2.7203e-01,\n",
       "            7.2757e-01, -4.8738e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.3263e-01, -3.9267e-02,  3.7338e-01,  ...,  5.6971e-02,\n",
       "           -7.3481e-01,  1.2074e-02],\n",
       "          [ 6.4466e-02, -3.4102e-01,  2.3846e-01,  ...,  1.5974e-01,\n",
       "           -8.4803e-01,  1.0083e-01],\n",
       "          [ 2.0168e-01, -3.8647e-01,  2.1557e-01,  ..., -1.1671e-01,\n",
       "           -8.0340e-01,  1.2341e-01],\n",
       "          ...,\n",
       "          [-1.4268e-01, -1.2399e-01,  5.5569e-01,  ...,  1.1445e-01,\n",
       "           -8.1405e-01,  4.1591e-01],\n",
       "          [-5.8878e-02, -2.7716e-01,  2.7189e-01,  ..., -9.3751e-02,\n",
       "           -1.1426e+00,  1.8464e-01],\n",
       "          [ 1.7431e-01,  4.6641e-02,  3.3055e-01,  ...,  3.7966e-01,\n",
       "           -8.8778e-01, -6.8483e-02]],\n",
       "\n",
       "         [[ 2.6972e-01, -1.0567e-01,  3.2027e-02,  ...,  5.3274e-02,\n",
       "           -3.9791e-04,  7.5771e-01],\n",
       "          [ 3.6542e-01, -8.8044e-02, -1.2865e-01,  ..., -3.4418e-02,\n",
       "           -8.8782e-02,  8.3374e-01],\n",
       "          [ 3.3415e-01, -5.4036e-03, -2.6894e-02,  ...,  4.4955e-02,\n",
       "            6.6554e-02,  7.1704e-01],\n",
       "          ...,\n",
       "          [ 3.3576e-01,  2.0495e-01, -1.5119e-01,  ...,  2.9721e-03,\n",
       "            3.7814e-02,  6.6385e-01],\n",
       "          [ 3.9029e-01,  2.4744e-02,  1.1299e-01,  ..., -1.0333e-01,\n",
       "            6.0536e-02,  9.4157e-01],\n",
       "          [ 4.0400e-01, -2.0606e-01, -1.7727e-01,  ..., -3.4581e-01,\n",
       "           -3.2783e-01,  7.6785e-01]],\n",
       "\n",
       "         [[ 1.7048e-01,  1.8170e-01, -1.8107e-01,  ...,  2.1516e-01,\n",
       "           -3.4139e-02, -5.7415e-02],\n",
       "          [ 7.4706e-02,  2.0320e-01,  2.3542e-01,  ..., -1.4815e-01,\n",
       "            2.9955e-01, -2.6272e-02],\n",
       "          [ 1.2199e-01,  2.5660e-01,  5.9935e-02,  ..., -6.9299e-04,\n",
       "            1.6163e-01, -6.8735e-02],\n",
       "          ...,\n",
       "          [ 1.0321e-01, -7.9846e-02,  1.0145e-01,  ...,  2.1556e-02,\n",
       "            4.3147e-02, -1.2106e-02],\n",
       "          [ 2.2976e-01,  1.6207e-01, -1.1506e-01,  ..., -7.1395e-03,\n",
       "           -6.1335e-02,  2.0166e-02],\n",
       "          [ 4.5113e-01,  3.0786e-01, -4.6971e-04,  ...,  1.9170e-01,\n",
       "            5.3457e-01,  1.5991e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.0852e-02,  2.5963e-01, -1.2263e-01,  ..., -2.0475e-01,\n",
       "           -1.9263e-01,  1.2030e-01],\n",
       "          [ 7.1267e-02,  5.0086e-01, -1.2690e-02,  ..., -2.5323e-01,\n",
       "           -1.4423e-01,  1.8232e-01],\n",
       "          [ 7.6885e-02,  4.7053e-01,  3.9130e-02,  ..., -2.6519e-01,\n",
       "           -1.8747e-01,  6.6554e-02],\n",
       "          ...,\n",
       "          [ 4.6391e-02,  5.0454e-01,  3.2167e-01,  ..., -1.4574e-01,\n",
       "           -2.7189e-01, -5.1019e-03],\n",
       "          [ 2.8304e-01,  3.7935e-01, -3.8711e-03,  ..., -3.1987e-01,\n",
       "           -2.3515e-01, -4.6044e-03],\n",
       "          [ 2.0223e-01,  6.9969e-01,  3.1078e-02,  ..., -3.9040e-01,\n",
       "           -3.3773e-01,  1.0879e-01]],\n",
       "\n",
       "         [[-3.1188e-01, -3.0358e-01, -1.5752e-01,  ..., -2.5955e-01,\n",
       "            6.4670e-02,  6.5291e-02],\n",
       "          [-2.4684e-01, -1.2837e-01, -2.3765e-02,  ..., -3.7005e-02,\n",
       "            1.9337e-01, -2.2531e-02],\n",
       "          [-1.5347e-01, -2.5219e-02, -8.8718e-02,  ..., -3.4438e-02,\n",
       "           -6.1267e-03, -1.1500e-01],\n",
       "          ...,\n",
       "          [-4.3200e-02, -5.7687e-02, -3.0639e-01,  ..., -7.0635e-02,\n",
       "           -3.5816e-02, -7.3593e-03],\n",
       "          [-1.4449e-01, -1.8206e-01,  5.4692e-02,  ..., -1.6222e-01,\n",
       "           -6.1402e-02, -2.6590e-02],\n",
       "          [-4.6061e-02,  4.0430e-02,  9.6205e-02,  ..., -2.7158e-01,\n",
       "            2.3319e-03, -2.1240e-01]],\n",
       "\n",
       "         [[ 1.5612e-02,  9.1012e-02,  1.5828e-01,  ...,  1.6083e-01,\n",
       "            2.2340e-01, -4.3663e-01],\n",
       "          [-6.7860e-02,  5.7925e-02,  2.2771e-02,  ...,  2.8538e-01,\n",
       "            1.8686e-01, -6.5796e-01],\n",
       "          [-6.2693e-02,  6.6696e-02,  9.7909e-02,  ...,  2.2029e-01,\n",
       "            4.0548e-01, -6.2012e-01],\n",
       "          ...,\n",
       "          [-4.7349e-02,  1.0376e-01, -2.1302e-02,  ...,  5.7560e-01,\n",
       "            4.2356e-01, -4.6172e-01],\n",
       "          [ 3.9949e-02,  4.5944e-02,  2.8310e-01,  ...,  4.0581e-01,\n",
       "            5.0133e-01, -5.0490e-01],\n",
       "          [ 2.1490e-01, -1.6650e-01,  5.1041e-01,  ...,  2.0466e-01,\n",
       "            3.6183e-01, -3.6910e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-3.0983e+00,  7.9435e-01, -3.8813e-01,  ..., -1.9726e-01,\n",
       "            7.4684e-01, -1.9241e+00],\n",
       "          [-1.5622e+00, -1.0046e+00, -7.1929e-01,  ...,  6.1725e-01,\n",
       "            4.8719e+00,  1.6120e+00],\n",
       "          [-1.1041e+00, -9.4147e-01,  1.4655e+00,  ...,  1.8205e+00,\n",
       "            2.6495e+00,  1.2626e+00],\n",
       "          ...,\n",
       "          [-1.2350e-01,  2.8935e-01,  2.3550e-01,  ...,  1.9526e+00,\n",
       "            3.9910e+00,  2.3879e+00],\n",
       "          [ 9.2520e-01, -2.4745e-01,  1.2404e+00,  ...,  2.4663e-01,\n",
       "            1.4804e+00,  1.2867e+00],\n",
       "          [-1.8797e+00,  1.3237e+00, -6.0817e-01,  ..., -2.6878e-01,\n",
       "           -2.3800e-01, -9.0329e-02]],\n",
       "\n",
       "         [[ 9.8675e-01, -2.2623e+00,  2.0685e+00,  ...,  2.2672e+00,\n",
       "           -5.7849e-01, -2.9876e+00],\n",
       "          [ 3.3038e-01, -9.3248e-01,  5.6146e-01,  ..., -7.1569e-01,\n",
       "           -2.1115e+00, -3.7927e+00],\n",
       "          [-2.1255e-02, -9.4794e-03,  8.3298e-01,  ..., -6.2962e-02,\n",
       "           -1.3179e+00, -4.1652e+00],\n",
       "          ...,\n",
       "          [ 1.0931e+00, -2.3549e-01, -3.3292e-01,  ...,  1.0518e+00,\n",
       "           -2.9925e+00, -1.9797e+00],\n",
       "          [ 7.6506e-01, -5.9219e-01,  5.6967e-01,  ...,  1.6004e-02,\n",
       "           -1.4849e+00, -3.5824e+00],\n",
       "          [ 4.3665e-01, -2.2661e+00, -3.9677e+00,  ..., -5.9975e-01,\n",
       "            3.8939e+00, -1.0809e+00]],\n",
       "\n",
       "         [[ 7.9639e-02,  8.0212e-01, -1.2447e+00,  ...,  1.2026e+00,\n",
       "            2.4686e-01,  1.1621e+00],\n",
       "          [ 1.4262e+00, -2.1493e+00, -6.3479e-01,  ..., -5.0744e-01,\n",
       "           -1.5749e+00,  1.8040e+00],\n",
       "          [ 1.0543e+00, -1.2555e-02,  1.2998e-01,  ..., -3.7213e-02,\n",
       "           -9.6843e-01,  6.2055e-01],\n",
       "          ...,\n",
       "          [-1.6240e+00, -9.3849e-01, -5.0726e-01,  ..., -2.0411e+00,\n",
       "           -7.8723e-01, -7.3547e-02],\n",
       "          [ 5.4018e-03, -6.5460e-02, -1.2519e+00,  ..., -6.1197e-01,\n",
       "            5.0605e-01, -9.9447e-01],\n",
       "          [-1.6554e+00, -1.5621e+00,  1.8502e+00,  ..., -2.8791e+00,\n",
       "           -3.0086e-01, -4.6969e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.4741e+00,  2.7656e+00,  1.0980e+00,  ...,  1.0862e+00,\n",
       "            7.7511e-02, -1.4610e+00],\n",
       "          [ 2.5827e+00,  1.1315e+00,  1.0657e+00,  ..., -3.7679e+00,\n",
       "           -2.5016e+00, -8.3485e-01],\n",
       "          [ 1.0448e+00,  3.6709e+00,  2.2994e+00,  ..., -1.2186e+00,\n",
       "            5.8465e-01, -3.5402e+00],\n",
       "          ...,\n",
       "          [-6.8722e-01,  1.9222e+00,  3.5443e-01,  ..., -6.2979e-01,\n",
       "           -6.4760e-01,  2.8214e+00],\n",
       "          [-1.0801e+00,  7.3658e-01,  1.4900e+00,  ..., -2.1131e-01,\n",
       "            1.7049e+00,  4.7856e-01],\n",
       "          [ 6.6412e-01, -8.4573e-01,  2.6625e+00,  ...,  1.6298e+00,\n",
       "            2.8177e+00,  8.5895e-01]],\n",
       "\n",
       "         [[-2.3375e+00, -6.8777e-01, -1.7436e+00,  ..., -7.8144e-01,\n",
       "           -3.8557e-01,  2.4584e-01],\n",
       "          [-1.3061e+00,  7.6634e-03,  3.9423e-01,  ...,  1.6409e-01,\n",
       "           -3.8397e-01,  1.0850e+00],\n",
       "          [-9.2993e-01,  4.2782e-01, -6.7582e-01,  ...,  4.3283e-02,\n",
       "           -8.3086e-01, -1.2332e-01],\n",
       "          ...,\n",
       "          [ 5.9874e-01, -1.1302e-01,  2.9274e-01,  ..., -1.6997e-01,\n",
       "           -4.9093e-01,  7.2453e-01],\n",
       "          [-6.2400e-01,  1.1575e-01,  6.5299e-01,  ..., -1.9915e+00,\n",
       "           -1.3018e+00, -1.3761e+00],\n",
       "          [ 1.2293e+00,  1.0224e+00,  2.6653e+00,  ..., -1.2547e-01,\n",
       "           -3.7331e-01, -1.7793e+00]],\n",
       "\n",
       "         [[-1.4098e+00,  1.5286e+00,  1.0912e-01,  ..., -1.1795e+00,\n",
       "            1.6999e+00,  1.9807e-02],\n",
       "          [-3.2108e-01,  1.6009e-01, -1.4020e+00,  ..., -6.1045e-01,\n",
       "            2.0771e+00, -5.6079e-01],\n",
       "          [ 4.4968e-01, -2.7624e-01, -7.6411e-01,  ..., -7.1276e-01,\n",
       "            6.0973e-01,  3.4574e-01],\n",
       "          ...,\n",
       "          [-1.3842e-01, -1.4273e+00, -1.7067e+00,  ..., -4.5123e-01,\n",
       "           -7.2356e-01, -1.1356e+00],\n",
       "          [ 1.8561e-01, -4.3629e-01, -6.6582e-01,  ..., -2.5027e+00,\n",
       "            4.2028e-01,  2.7610e-01],\n",
       "          [ 2.1075e+00,  7.3671e-01, -1.4612e+00,  ..., -1.4710e+00,\n",
       "           -4.0066e-01,  1.7549e+00]]],\n",
       "\n",
       "\n",
       "        [[[-2.3893e+00,  2.4866e+00, -3.5052e+00,  ..., -1.4932e+00,\n",
       "            5.7456e-01,  2.1705e+00],\n",
       "          [-4.2620e-01, -3.0668e-01,  5.8592e-01,  ...,  5.9108e-01,\n",
       "            1.9505e+00,  1.4621e+00],\n",
       "          [-9.3426e-01, -1.5676e+00, -7.7335e-01,  ..., -2.3996e+00,\n",
       "            2.0461e+00,  8.8200e-01],\n",
       "          ...,\n",
       "          [-1.1621e+00, -1.6373e+00,  8.6277e-01,  ...,  2.6128e+00,\n",
       "            2.9208e+00,  2.3867e+00],\n",
       "          [-5.1900e-01,  5.1989e-01,  1.5763e+00,  ...,  1.9124e+00,\n",
       "            1.8086e+00, -3.9875e-03],\n",
       "          [-1.9670e+00,  1.2927e+00, -5.8251e-01,  ..., -2.2688e-01,\n",
       "           -2.4820e-01, -1.2672e-01]],\n",
       "\n",
       "         [[ 6.7698e-01, -8.0790e-01, -2.3975e+00,  ...,  1.6870e+00,\n",
       "           -2.2304e+00, -7.3004e-01],\n",
       "          [ 3.9011e-01,  2.5454e-01,  2.0542e-03,  ...,  8.5155e-01,\n",
       "           -2.4215e+00, -3.1663e+00],\n",
       "          [ 2.0927e+00, -1.1180e-01, -1.4781e+00,  ...,  5.0992e-01,\n",
       "           -1.5393e+00, -2.8074e+00],\n",
       "          ...,\n",
       "          [-6.7113e-01, -8.9246e-01,  2.4250e+00,  ...,  1.8765e+00,\n",
       "            7.5860e-01, -8.7180e-01],\n",
       "          [-4.2665e-01, -9.2635e-01,  9.3208e-01,  ...,  2.1697e-01,\n",
       "            1.5819e+00, -1.5109e+00],\n",
       "          [ 3.5362e-01, -2.2796e+00, -3.9908e+00,  ..., -6.1042e-01,\n",
       "            3.9976e+00, -1.0151e+00]],\n",
       "\n",
       "         [[-1.1547e+00,  4.9837e-02,  1.3175e+00,  ..., -1.7142e+00,\n",
       "           -4.2422e-01,  1.0870e+00],\n",
       "          [-1.4896e-01,  1.0088e+00, -2.1990e-01,  ...,  2.9151e-01,\n",
       "            1.7909e-02, -2.5394e-01],\n",
       "          [ 8.3402e-01, -1.2002e+00, -8.8634e-01,  ...,  9.0448e-01,\n",
       "            1.3962e+00,  6.3891e-01],\n",
       "          ...,\n",
       "          [-2.8369e-01, -2.0037e-01, -1.2421e+00,  ...,  2.2442e+00,\n",
       "           -1.2712e+00,  2.1449e-01],\n",
       "          [ 1.5291e-01,  2.4926e-01, -7.5129e-01,  ...,  5.1047e-02,\n",
       "           -1.2489e+00, -4.0727e-01],\n",
       "          [-1.6465e+00, -1.5675e+00,  1.8706e+00,  ..., -2.8381e+00,\n",
       "           -3.8019e-01, -5.2503e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.9014e-01, -1.3342e+00,  5.7082e-01,  ...,  1.1120e+00,\n",
       "            7.7712e-01,  1.1525e+00],\n",
       "          [-5.2873e-01,  4.4445e+00,  1.7912e+00,  ...,  1.0041e-01,\n",
       "            2.5921e+00, -1.6853e+00],\n",
       "          [-2.7329e+00, -1.5253e+00, -1.9307e+00,  ...,  1.5561e+00,\n",
       "            1.2052e+00, -2.8567e-01],\n",
       "          ...,\n",
       "          [-1.1705e+00, -1.3907e+00,  1.4283e+00,  ...,  5.0161e-01,\n",
       "           -2.8119e+00,  1.9430e-01],\n",
       "          [-1.6678e-01, -1.9506e+00,  1.6958e+00,  ..., -3.3384e-01,\n",
       "            1.3528e-01,  1.4941e+00],\n",
       "          [ 7.1921e-01, -8.9418e-01,  2.6676e+00,  ...,  1.6316e+00,\n",
       "            2.7898e+00,  9.1870e-01]],\n",
       "\n",
       "         [[ 1.7523e+00, -4.5012e-01, -1.4176e-01,  ..., -1.2649e+00,\n",
       "            7.4343e-01,  7.0807e-01],\n",
       "          [-3.5747e-01, -2.9570e-01, -1.6063e-01,  ..., -3.0508e-01,\n",
       "            1.3291e-01, -1.2467e+00],\n",
       "          [ 4.1428e-01,  8.1308e-01, -3.1609e-01,  ..., -9.5435e-01,\n",
       "            2.2704e-01, -6.1407e-02],\n",
       "          ...,\n",
       "          [ 1.8929e+00, -1.3938e+00, -1.5244e+00,  ..., -5.6892e-01,\n",
       "            2.0208e+00, -4.3837e-01],\n",
       "          [ 6.3910e-01, -1.6042e+00, -6.8313e-02,  ..., -6.0782e-01,\n",
       "           -1.8732e-01, -1.0193e+00],\n",
       "          [ 1.2670e+00,  9.6901e-01,  2.6515e+00,  ..., -1.9153e-02,\n",
       "           -3.4983e-01, -1.7503e+00]],\n",
       "\n",
       "         [[ 1.0334e+00, -6.3312e-02,  6.7447e-01,  ..., -6.2707e-01,\n",
       "           -3.3270e-01, -9.5207e-01],\n",
       "          [ 3.6838e-01, -6.4512e-01, -6.8223e-02,  ..., -9.6221e-01,\n",
       "            3.6847e-01, -7.5250e-01],\n",
       "          [ 9.9108e-01, -3.5018e-01, -2.5771e-03,  ..., -1.0724e+00,\n",
       "            2.1205e+00, -3.5882e+00],\n",
       "          ...,\n",
       "          [ 5.9767e-01,  5.6307e-01, -7.6049e-03,  ...,  3.5614e-01,\n",
       "            1.4405e-01,  7.9887e-02],\n",
       "          [ 4.8796e-01,  6.0101e-01, -9.7214e-01,  ..., -1.2938e+00,\n",
       "           -1.1532e+00,  1.3699e+00],\n",
       "          [ 2.1459e+00,  7.0429e-01, -1.4374e+00,  ..., -1.4925e+00,\n",
       "           -4.7216e-01,  1.8473e+00]]],\n",
       "\n",
       "\n",
       "        [[[-3.1824e+00, -1.8372e+00, -7.2064e-01,  ...,  9.4105e-01,\n",
       "            2.3617e+00,  3.6098e+00],\n",
       "          [-3.2042e+00,  5.1160e-02, -1.6299e+00,  ..., -1.7279e+00,\n",
       "            2.6345e+00,  9.9174e-01],\n",
       "          [-3.2227e+00,  2.2352e+00, -1.3579e+00,  ..., -3.7994e+00,\n",
       "            9.3385e-01, -8.4697e-02],\n",
       "          ...,\n",
       "          [ 7.5692e-01,  1.1659e+00, -7.1570e-01,  ..., -8.0251e-01,\n",
       "            8.8206e-01, -3.4440e-02],\n",
       "          [-1.3991e+00, -5.9112e-01, -1.6084e-01,  ...,  7.4473e-01,\n",
       "            1.6747e+00,  5.8344e-01],\n",
       "          [-1.9464e+00,  1.3103e+00, -6.1591e-01,  ..., -2.8930e-01,\n",
       "           -2.2274e-01, -1.1934e-01]],\n",
       "\n",
       "         [[ 2.0449e+00,  5.3198e-01,  1.5582e+00,  ...,  8.6112e-01,\n",
       "           -1.6262e-01, -1.6878e+00],\n",
       "          [ 3.8583e-01, -9.5490e-01,  1.8634e-01,  ...,  1.0372e+00,\n",
       "            1.9550e-01, -2.1119e+00],\n",
       "          [-3.6426e-01,  2.2504e+00, -1.6332e+00,  ...,  2.3452e+00,\n",
       "            6.7932e-01, -3.4872e+00],\n",
       "          ...,\n",
       "          [-7.9270e-01, -4.6973e-01, -2.1779e+00,  ...,  3.6448e-02,\n",
       "            1.1646e+00, -2.5385e+00],\n",
       "          [ 2.7136e-01, -7.4329e-01, -3.6694e-01,  ...,  1.1174e-01,\n",
       "           -8.7904e-01, -2.0033e+00],\n",
       "          [ 4.0646e-01, -2.2271e+00, -4.0211e+00,  ..., -5.7514e-01,\n",
       "            3.9545e+00, -9.8992e-01]],\n",
       "\n",
       "         [[-8.3920e-01, -1.2598e+00,  6.5074e-01,  ..., -3.6981e-01,\n",
       "           -9.8473e-01,  5.4552e-01],\n",
       "          [-3.3476e+00,  2.7962e+00, -1.4631e+00,  ...,  2.7943e-01,\n",
       "           -1.6784e-01, -2.2340e+00],\n",
       "          [-2.2602e+00,  2.7429e+00, -2.2893e+00,  ...,  1.3553e+00,\n",
       "           -1.3540e-01, -6.5331e-01],\n",
       "          ...,\n",
       "          [-1.3168e+00,  1.8566e-01, -1.3453e+00,  ..., -8.2396e-01,\n",
       "           -1.8329e+00, -7.9444e-01],\n",
       "          [-2.3114e+00, -5.7484e-01, -8.2158e-01,  ..., -4.4465e-01,\n",
       "            1.4743e-01, -9.3758e-01],\n",
       "          [-1.7027e+00, -1.5377e+00,  1.8093e+00,  ..., -2.8216e+00,\n",
       "           -3.5244e-01, -4.6090e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.4908e+00,  1.4910e+00, -1.3523e+00,  ...,  4.2173e-01,\n",
       "           -2.0625e-02, -3.2278e-02],\n",
       "          [ 9.0678e-01,  1.2926e+00, -1.3278e+00,  ...,  1.3545e+00,\n",
       "            1.3893e+00, -1.2913e+00],\n",
       "          [ 2.9893e-01, -1.9775e+00, -1.6563e+00,  ..., -1.4029e+00,\n",
       "           -2.1451e+00,  2.7188e+00],\n",
       "          ...,\n",
       "          [-1.9441e+00, -1.0014e+00, -4.7137e-01,  ...,  1.9787e+00,\n",
       "           -6.7636e-01, -9.0200e-02],\n",
       "          [-5.8774e-02, -4.9164e-01, -1.0299e+00,  ...,  1.3837e+00,\n",
       "            1.8065e+00,  1.5524e+00],\n",
       "          [ 7.3028e-01, -8.6617e-01,  2.6371e+00,  ...,  1.5941e+00,\n",
       "            2.8488e+00,  8.1616e-01]],\n",
       "\n",
       "         [[ 2.6882e+00, -1.1548e-01, -1.3837e+00,  ..., -4.5350e-01,\n",
       "            6.9889e-01,  2.0164e+00],\n",
       "          [-1.2739e+00,  3.5980e-01, -1.2414e+00,  ..., -1.3025e+00,\n",
       "            9.1086e-01,  3.6178e-01],\n",
       "          [-1.5184e-01, -1.9062e+00, -1.0629e+00,  ...,  4.9586e-01,\n",
       "            1.2846e+00,  6.8072e-01],\n",
       "          ...,\n",
       "          [ 6.8351e-01,  1.1969e-01, -5.2571e-02,  ...,  1.2166e-01,\n",
       "            2.8128e-01, -1.0545e+00],\n",
       "          [ 1.6463e+00, -7.4002e-01, -1.0002e+00,  ..., -8.7306e-01,\n",
       "            1.3957e-01, -1.2826e+00],\n",
       "          [ 1.2577e+00,  9.6713e-01,  2.6461e+00,  ..., -8.9453e-02,\n",
       "           -3.6497e-01, -1.7484e+00]],\n",
       "\n",
       "         [[-2.1262e-01,  1.5174e+00, -3.3956e-02,  ...,  7.9658e-01,\n",
       "            1.4950e+00, -1.0791e+00],\n",
       "          [ 5.6807e-02,  1.5626e-01, -3.2631e-02,  ..., -3.6809e+00,\n",
       "            1.8688e+00, -1.8175e-01],\n",
       "          [ 2.9662e-01,  7.1077e-01, -3.7644e-01,  ..., -1.0568e+00,\n",
       "            1.6180e+00, -6.0256e-02],\n",
       "          ...,\n",
       "          [ 3.1694e-01, -2.9287e-03, -7.2335e-02,  ..., -2.0813e+00,\n",
       "           -1.1779e+00,  1.0533e+00],\n",
       "          [ 4.9923e-01,  1.0352e+00,  3.7192e-01,  ..., -2.5669e+00,\n",
       "           -2.2018e-01, -1.4690e+00],\n",
       "          [ 2.1533e+00,  7.3207e-01, -1.4111e+00,  ..., -1.4706e+00,\n",
       "           -4.0274e-01,  1.7863e+00]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 9.9698e-02, -2.0299e+00,  3.3719e-01,  ..., -1.1275e+00,\n",
       "            1.5426e+00,  5.7784e-02],\n",
       "          [-1.8898e+00,  3.8188e-01,  8.3294e-01,  ..., -7.7425e-01,\n",
       "            1.1549e+00,  1.5039e+00],\n",
       "          [-8.5260e-01, -1.2255e+00,  7.7016e-01,  ..., -5.9772e-01,\n",
       "            2.2425e+00,  1.0807e+00],\n",
       "          ...,\n",
       "          [-1.1510e+00,  8.0110e-01,  4.9409e-01,  ..., -2.1548e-01,\n",
       "            2.1675e+00,  2.1755e+00],\n",
       "          [ 1.4908e-01, -3.4025e-01, -5.0712e-01,  ..., -7.3682e-01,\n",
       "           -2.7064e-01,  5.6152e-01],\n",
       "          [-4.0067e-01,  4.0599e-01, -1.0984e-01,  ...,  2.7060e-01,\n",
       "           -6.3232e-01, -1.3936e-01]],\n",
       "\n",
       "         [[ 3.6541e-01,  8.5992e-01,  5.4679e-02,  ..., -3.8570e-01,\n",
       "           -6.5037e-01,  2.5878e-02],\n",
       "          [-3.7158e-01, -1.1991e+00, -6.8507e-01,  ...,  1.7219e+00,\n",
       "           -1.3372e-01, -6.2035e-01],\n",
       "          [-6.9849e-01, -1.3189e+00, -2.4040e+00,  ...,  1.6478e+00,\n",
       "            1.2284e+00,  7.4652e-01],\n",
       "          ...,\n",
       "          [ 1.8698e-01, -4.7000e-02, -6.7072e-01,  ...,  1.2699e+00,\n",
       "           -6.1272e-01, -1.6160e+00],\n",
       "          [ 7.3042e-01, -1.0849e+00, -5.1714e-01,  ...,  1.1833e-02,\n",
       "           -5.7380e-02, -9.5766e-01],\n",
       "          [ 2.8546e-01, -4.7631e-01, -3.0094e-01,  ...,  5.9378e-01,\n",
       "            1.5544e+00,  2.9501e-01]],\n",
       "\n",
       "         [[ 1.1226e+00,  1.1721e-01, -8.5585e-01,  ...,  9.7939e-01,\n",
       "           -2.6482e-01, -2.4999e+00],\n",
       "          [-4.1527e-01,  1.1103e+00, -6.8513e-01,  ..., -2.3939e-01,\n",
       "            8.9534e-01, -2.1182e+00],\n",
       "          [ 3.9410e-01,  5.8908e-01,  3.1479e-01,  ...,  2.0906e-01,\n",
       "            2.0694e+00, -1.8502e+00],\n",
       "          ...,\n",
       "          [ 1.1333e+00,  3.7956e-01, -1.3193e+00,  ...,  7.2871e-02,\n",
       "            1.8592e+00, -1.2932e+00],\n",
       "          [ 1.1243e+00,  1.5725e+00, -2.4084e-01,  ..., -5.0725e-01,\n",
       "            6.1344e-01, -2.2256e+00],\n",
       "          [ 8.2717e-01,  8.5762e-02,  2.1409e-02,  ...,  1.3618e-01,\n",
       "            7.5292e-01, -8.9564e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-2.6432e-02, -4.5177e-01,  8.2738e-01,  ..., -1.2693e+00,\n",
       "           -2.5293e+00,  2.1419e+00],\n",
       "          [ 6.2308e-01,  5.2605e-01, -1.0615e+00,  ...,  1.0420e+00,\n",
       "           -2.3809e+00, -5.0698e-01],\n",
       "          [-6.5036e-01,  6.5175e-01,  1.0021e+00,  ...,  1.0298e+00,\n",
       "           -1.0346e+00, -8.4500e-01],\n",
       "          ...,\n",
       "          [-2.0313e-01,  5.0175e-01, -1.1152e+00,  ...,  7.4222e-01,\n",
       "           -1.3629e+00, -6.5924e-01],\n",
       "          [-3.5257e-01, -4.9452e-01, -2.7021e-01,  ...,  9.2177e-01,\n",
       "           -2.7040e-01, -3.7239e-01],\n",
       "          [ 4.4487e-02,  2.5771e-01, -2.7964e-01,  ..., -6.4003e-01,\n",
       "            2.3931e-01, -5.8547e-01]],\n",
       "\n",
       "         [[ 2.6292e-01, -4.8304e-01,  5.9942e-01,  ..., -5.8896e-02,\n",
       "            1.1631e+00, -6.8525e-01],\n",
       "          [-1.0390e-01,  8.7844e-01, -1.1987e+00,  ...,  4.7147e-01,\n",
       "            6.7015e-01, -6.5048e-01],\n",
       "          [-1.2289e+00,  9.2800e-01, -2.2710e+00,  ..., -5.5477e-01,\n",
       "            2.1705e-01, -5.2859e-01],\n",
       "          ...,\n",
       "          [ 2.7788e-02,  5.8168e-01, -1.1816e+00,  ..., -9.7153e-01,\n",
       "            2.1958e+00, -2.5039e-01],\n",
       "          [ 2.9711e-01,  8.8671e-01, -1.7261e-01,  ..., -9.7237e-01,\n",
       "            7.0937e-01, -2.3843e-01],\n",
       "          [ 3.8824e-01, -1.2395e-01, -9.8727e-01,  ..., -9.8480e-01,\n",
       "            8.2940e-02, -7.2173e-01]],\n",
       "\n",
       "         [[ 7.7781e-01, -1.1853e+00, -1.0528e-01,  ...,  7.3427e-01,\n",
       "            2.8684e+00, -1.2740e+00],\n",
       "          [-2.0176e-01,  2.1948e-01,  7.7449e-02,  ..., -3.3221e-01,\n",
       "            6.8253e-01, -3.5519e-01],\n",
       "          [-1.1427e+00, -1.9074e-01,  3.6685e-02,  ...,  1.2098e+00,\n",
       "            9.5107e-01,  2.3064e-01],\n",
       "          ...,\n",
       "          [ 6.9874e-01,  1.6061e+00, -2.9882e-01,  ...,  6.5172e-01,\n",
       "            1.8035e-01, -1.1450e+00],\n",
       "          [-1.5027e-01, -3.4373e-01, -8.9007e-01,  ...,  8.3627e-01,\n",
       "            1.1286e+00, -7.4852e-01],\n",
       "          [ 8.0197e-01, -1.0404e-01,  6.6696e-01,  ...,  2.3757e-01,\n",
       "            8.1710e-01, -2.1526e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.6104e+00, -7.4305e-02,  1.2198e+00,  ..., -9.4441e-01,\n",
       "            1.5454e+00,  1.5148e+00],\n",
       "          [-6.3215e-01, -1.6732e+00,  2.8689e-01,  ..., -8.4548e-01,\n",
       "            3.0119e+00,  1.3262e+00],\n",
       "          [-2.3664e-01,  7.6074e-01,  6.8885e-03,  ..., -2.7799e-01,\n",
       "           -2.8413e+00,  1.6468e+00],\n",
       "          ...,\n",
       "          [-1.1078e+00, -1.6323e+00,  8.1593e-01,  ..., -3.2510e+00,\n",
       "            1.5262e+00,  3.2746e+00],\n",
       "          [-6.0783e-01, -3.2699e-02, -1.3610e-01,  ..., -7.4142e-01,\n",
       "            3.8153e-01,  2.1679e+00],\n",
       "          [-4.3603e-01,  4.1720e-01, -9.6380e-02,  ...,  2.6535e-01,\n",
       "           -5.8142e-01, -1.1940e-01]],\n",
       "\n",
       "         [[-2.7305e-01, -1.7460e+00, -8.9210e-01,  ...,  1.0372e+00,\n",
       "            2.8468e-01,  1.1494e+00],\n",
       "          [-8.0400e-01, -1.1844e+00, -2.3576e+00,  ...,  1.7246e+00,\n",
       "           -2.9438e-01,  7.0759e-01],\n",
       "          [ 3.7227e-01,  5.2560e-01,  3.5740e-01,  ...,  1.1669e+00,\n",
       "            5.4445e-01,  2.6690e-01],\n",
       "          ...,\n",
       "          [-1.4940e+00,  8.9876e-01,  3.1052e-01,  ...,  1.5523e+00,\n",
       "            9.8677e-01,  1.7962e+00],\n",
       "          [-7.6738e-02, -1.2225e-01,  2.6352e-01,  ..., -1.9597e-01,\n",
       "            9.2656e-01, -2.2712e-01],\n",
       "          [ 2.4478e-01, -4.8073e-01, -2.6273e-01,  ...,  5.7371e-01,\n",
       "            1.5078e+00,  2.7300e-01]],\n",
       "\n",
       "         [[ 1.1625e-01,  3.9331e-01,  9.0181e-01,  ..., -1.8459e-01,\n",
       "            2.8265e+00,  1.5239e+00],\n",
       "          [ 4.5673e-01,  5.5176e-02,  9.8260e-01,  ..., -4.7334e-01,\n",
       "            1.2515e+00, -2.1146e+00],\n",
       "          [ 4.0082e-01,  6.1772e-01,  7.5519e-01,  ..., -2.8610e-01,\n",
       "            1.7355e-01, -7.0172e-01],\n",
       "          ...,\n",
       "          [ 1.6509e+00,  1.0476e+00, -3.7973e-01,  ..., -7.9120e-01,\n",
       "            2.5770e-01, -3.4189e-01],\n",
       "          [ 1.6808e+00,  4.4354e-01, -4.0133e-01,  ..., -7.1860e-01,\n",
       "            1.2566e+00, -1.4706e+00],\n",
       "          [ 8.4049e-01,  8.2715e-02, -6.3599e-03,  ...,  1.0442e-01,\n",
       "            7.4943e-01, -8.9331e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-4.3208e-01, -8.8680e-01, -3.0062e-01,  ...,  7.3024e-01,\n",
       "           -1.9891e+00,  5.8041e-01],\n",
       "          [-1.0917e+00,  5.4600e-01,  1.6964e+00,  ...,  1.0781e+00,\n",
       "           -7.3813e-01, -5.5684e-01],\n",
       "          [-1.2465e+00,  4.9180e-01,  1.1442e+00,  ...,  1.3569e-01,\n",
       "            4.4499e-01, -1.0853e+00],\n",
       "          ...,\n",
       "          [-6.6597e-01, -5.9741e-01,  1.4107e+00,  ...,  2.4949e-01,\n",
       "           -1.5002e+00, -4.9221e-01],\n",
       "          [ 1.5869e-01, -2.1663e-01, -2.2691e-01,  ...,  2.9952e-01,\n",
       "           -1.0835e+00, -2.5749e-01],\n",
       "          [ 8.1052e-02,  2.7305e-01, -3.2664e-01,  ..., -5.7979e-01,\n",
       "            1.9157e-01, -6.1537e-01]],\n",
       "\n",
       "         [[ 1.3531e+00,  1.8298e+00,  1.7216e-01,  ..., -1.7831e+00,\n",
       "           -1.0896e+00, -1.8751e-01],\n",
       "          [-1.2168e+00,  4.0790e-01, -1.6825e+00,  ..., -1.1866e+00,\n",
       "            3.6799e-01, -6.3197e-01],\n",
       "          [-1.4766e+00,  5.9418e-01,  1.1290e+00,  ...,  6.3193e-01,\n",
       "           -8.9091e-01, -3.7442e+00],\n",
       "          ...,\n",
       "          [-5.3981e-01, -6.2822e-02, -1.3928e-01,  ..., -8.6996e-01,\n",
       "           -2.7095e-01, -7.1771e-02],\n",
       "          [ 2.9975e-01,  5.9089e-01, -7.8984e-01,  ..., -1.8745e-01,\n",
       "            1.2558e-01,  9.5301e-03],\n",
       "          [ 3.8151e-01, -1.3681e-01, -9.9681e-01,  ..., -9.6933e-01,\n",
       "            1.0537e-01, -7.0935e-01]],\n",
       "\n",
       "         [[-5.8794e-01,  7.3350e-01, -1.2569e+00,  ..., -6.9102e-01,\n",
       "           -1.5116e+00, -5.0615e-01],\n",
       "          [-4.2630e-01,  1.2753e+00, -9.5438e-02,  ...,  4.9098e-01,\n",
       "            4.5145e-01, -6.8263e-01],\n",
       "          [ 1.4317e-01,  7.0270e-01, -5.7886e-01,  ...,  9.9154e-01,\n",
       "           -3.4337e-01, -1.7685e+00],\n",
       "          ...,\n",
       "          [ 2.9892e-01, -1.1522e-02, -1.9654e-02,  ...,  6.4274e-01,\n",
       "            7.8261e-05,  4.4490e-01],\n",
       "          [-3.8896e-01,  1.5999e-01, -7.0344e-01,  ..., -5.8050e-01,\n",
       "            2.1092e+00, -9.6827e-01],\n",
       "          [ 7.8155e-01, -8.1824e-02,  7.1432e-01,  ...,  1.8059e-01,\n",
       "            8.6368e-01, -1.6063e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.5846e+00, -2.6961e+00, -2.7228e+00,  ...,  4.1686e-01,\n",
       "            2.5043e+00,  2.1094e+00],\n",
       "          [-4.8863e-01,  3.3138e-01, -2.2796e-01,  ..., -9.8019e-01,\n",
       "            1.1958e+00,  1.5928e+00],\n",
       "          [-1.7774e+00, -1.1659e+00,  3.1364e-01,  ..., -7.8058e-01,\n",
       "            2.1013e+00,  1.4055e+00],\n",
       "          ...,\n",
       "          [ 1.2345e+00,  1.6264e-01, -5.0661e-01,  ...,  1.5956e+00,\n",
       "            9.5532e-01,  2.8744e+00],\n",
       "          [-4.4355e-01, -3.4659e-01, -6.1485e-01,  ..., -1.3921e+00,\n",
       "           -5.2603e-01,  1.1640e+00],\n",
       "          [-4.0830e-01,  4.1831e-01, -9.4453e-02,  ...,  2.9179e-01,\n",
       "           -6.3024e-01, -1.1954e-01]],\n",
       "\n",
       "         [[-2.5184e-01, -7.4857e-01,  6.9889e-01,  ...,  2.2406e+00,\n",
       "           -1.5716e-01,  9.9419e-01],\n",
       "          [-1.1259e+00, -8.1699e-02,  1.2327e-01,  ...,  9.9404e-01,\n",
       "            1.1650e+00, -3.6690e-01],\n",
       "          [-1.1905e+00, -1.0240e+00,  8.7588e-01,  ...,  1.0332e+00,\n",
       "            2.4004e-02, -1.6679e-01],\n",
       "          ...,\n",
       "          [ 6.5504e-01,  5.9774e-01, -2.7019e-01,  ..., -1.2168e-01,\n",
       "            1.3660e+00, -9.3437e-01],\n",
       "          [-1.2939e-01, -5.9944e-01,  5.6980e-01,  ..., -2.3089e-01,\n",
       "           -4.2957e-02, -2.3114e-01],\n",
       "          [ 2.4723e-01, -4.6020e-01, -2.3513e-01,  ...,  5.4596e-01,\n",
       "            1.5678e+00,  2.9461e-01]],\n",
       "\n",
       "         [[ 8.6917e-01, -1.0570e+00,  5.8305e-01,  ...,  3.4872e-02,\n",
       "            2.1740e+00, -8.5218e-01],\n",
       "          [ 1.9885e+00, -1.2389e-02, -4.1142e-01,  ..., -3.6115e-01,\n",
       "            9.0416e-01, -2.1703e+00],\n",
       "          [ 7.4586e-01, -1.0429e+00,  2.4073e+00,  ..., -5.1782e-01,\n",
       "            9.4357e-01, -1.3853e+00],\n",
       "          ...,\n",
       "          [ 1.8514e-01,  1.1138e+00, -7.4719e-01,  ..., -2.2546e-01,\n",
       "            9.4396e-01, -9.7694e-01],\n",
       "          [ 1.5174e+00,  3.6610e-01,  1.8754e-01,  ...,  6.7591e-01,\n",
       "            6.5079e-01, -6.5163e-01],\n",
       "          [ 8.3063e-01,  7.5936e-02,  2.6709e-02,  ...,  1.2218e-01,\n",
       "            7.4539e-01, -8.6380e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.1523e-01,  1.0500e+00,  9.5710e-01,  ...,  4.2378e-01,\n",
       "           -1.7334e+00, -8.9624e-01],\n",
       "          [ 5.4673e-01, -7.9345e-02,  3.8583e-01,  ...,  4.3633e-01,\n",
       "           -2.2786e+00,  5.2721e-01],\n",
       "          [-1.3456e+00,  1.7058e+00,  1.7429e+00,  ...,  3.2540e-01,\n",
       "           -1.3404e+00, -6.4876e-01],\n",
       "          ...,\n",
       "          [-9.9422e-01,  8.2836e-01,  5.7181e-01,  ..., -4.5451e-01,\n",
       "           -3.0251e-01,  1.8070e+00],\n",
       "          [-3.0934e-01,  9.6789e-02,  1.0971e-01,  ...,  7.2158e-01,\n",
       "           -6.6438e-01, -2.6671e-01],\n",
       "          [ 2.6614e-02,  2.5339e-01, -3.3028e-01,  ..., -6.2682e-01,\n",
       "            2.2533e-01, -5.9334e-01]],\n",
       "\n",
       "         [[-8.7164e-01,  1.2788e+00, -5.7241e-01,  ...,  7.3040e-01,\n",
       "            5.8121e-02, -9.4244e-01],\n",
       "          [ 1.0704e+00,  6.1256e-01, -1.6922e+00,  ...,  1.3395e+00,\n",
       "           -4.9866e-01, -1.7248e+00],\n",
       "          [ 2.6019e-01, -4.6755e-01,  1.4529e-01,  ..., -4.1786e-01,\n",
       "           -8.6386e-01, -1.8692e+00],\n",
       "          ...,\n",
       "          [ 7.2138e-01, -4.0410e-02, -9.8701e-01,  ..., -7.4633e-01,\n",
       "            1.9366e+00, -2.1686e-01],\n",
       "          [-1.4424e-01,  1.1335e-01, -6.7419e-01,  ..., -5.6296e-01,\n",
       "            8.8325e-01, -8.4293e-01],\n",
       "          [ 4.2511e-01, -1.4264e-01, -9.9652e-01,  ..., -1.0052e+00,\n",
       "            9.9745e-02, -6.9186e-01]],\n",
       "\n",
       "         [[-1.0857e+00,  1.0427e+00,  1.7880e-01,  ...,  1.4823e-02,\n",
       "            5.9645e-01, -8.3849e-01],\n",
       "          [-5.9231e-01, -4.7098e-02,  3.4093e-01,  ..., -6.4109e-01,\n",
       "            2.4963e-01,  6.4005e-01],\n",
       "          [ 3.3619e-01, -1.1880e+00,  6.3410e-01,  ..., -5.2456e-01,\n",
       "            9.9191e-01, -7.9714e-01],\n",
       "          ...,\n",
       "          [ 9.8944e-02,  3.8607e-01,  2.0828e-01,  ..., -7.0342e-01,\n",
       "            9.9059e-01, -9.3156e-01],\n",
       "          [-1.8494e-01, -5.7973e-01, -5.9668e-01,  ...,  5.2015e-01,\n",
       "           -1.8931e-01, -4.4088e-01],\n",
       "          [ 8.3064e-01, -1.0098e-01,  6.7382e-01,  ...,  2.1924e-01,\n",
       "            8.2670e-01, -2.0539e-01]]]], grad_fn=<TransposeBackward0>)), (tensor([[[[ 3.5117e-01,  6.1535e-02,  8.3281e-01,  ..., -2.9493e-01,\n",
       "            6.0491e-01,  1.7189e-01],\n",
       "          [-4.1537e-02,  1.4005e-01,  5.1025e-01,  ...,  1.9439e-01,\n",
       "            8.4228e-01, -5.4868e-01],\n",
       "          [-4.6016e-02,  1.4921e-01,  3.5819e-01,  ...,  3.3227e-01,\n",
       "            6.7042e-01, -4.8772e-01],\n",
       "          ...,\n",
       "          [ 2.2005e-01,  3.1549e-01,  7.7989e-01,  ...,  2.4432e-01,\n",
       "            5.6107e-01, -2.0748e-01],\n",
       "          [ 8.7816e-02,  2.4428e-01,  5.1851e-01,  ...,  1.3264e-01,\n",
       "            6.2961e-01, -2.0518e-01],\n",
       "          [ 4.9260e-01,  4.3024e-01,  5.3177e-01,  ..., -1.1314e-01,\n",
       "            5.0777e-01, -2.9994e-02]],\n",
       "\n",
       "         [[ 7.0607e-01, -1.0735e+00,  8.9172e-02,  ..., -2.9649e-01,\n",
       "           -1.1968e+00,  3.1630e-01],\n",
       "          [ 4.9450e-01, -5.1770e-01,  6.1931e-02,  ..., -4.7834e-01,\n",
       "           -4.8152e-01,  7.8360e-02],\n",
       "          [ 1.2601e-01, -3.3172e-01,  4.9063e-02,  ..., -5.9152e-01,\n",
       "           -3.8913e-01, -1.4922e-02],\n",
       "          ...,\n",
       "          [ 3.8917e-01, -3.9429e-01, -3.4417e-01,  ..., -5.2189e-01,\n",
       "           -1.0903e-01,  1.1980e-01],\n",
       "          [ 3.3821e-01, -4.9706e-01, -3.7851e-02,  ..., -3.4331e-01,\n",
       "           -3.4211e-01,  1.1495e-01],\n",
       "          [ 4.9135e-01, -4.6375e-01, -8.9240e-02,  ..., -2.9331e-01,\n",
       "           -4.7880e-01,  3.9243e-02]],\n",
       "\n",
       "         [[ 1.8880e-01, -3.1255e-01, -1.3007e-01,  ...,  5.2747e-01,\n",
       "            4.2702e-01, -1.4015e-01],\n",
       "          [ 1.3769e-01, -1.8248e-01, -2.7950e-01,  ...,  2.0696e-01,\n",
       "            2.2564e-01, -2.0028e-01],\n",
       "          [ 1.2640e-01,  9.0725e-02, -1.2585e-01,  ...,  1.0732e-01,\n",
       "            3.3072e-01, -7.9035e-02],\n",
       "          ...,\n",
       "          [ 3.3869e-01, -3.9930e-02, -1.9603e-01,  ...,  2.9623e-01,\n",
       "            1.7055e-01, -1.5876e-01],\n",
       "          [ 1.9289e-01,  4.6469e-02, -3.2058e-01,  ...,  3.6782e-01,\n",
       "            9.1449e-02, -6.6071e-02],\n",
       "          [-2.0329e-02, -7.1898e-04, -1.0878e-01,  ...,  2.2133e-01,\n",
       "            1.8666e-01, -4.1068e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.7504e-01, -3.9878e-01, -2.8772e-01,  ..., -8.3544e-01,\n",
       "            6.7166e-01, -7.8920e-01],\n",
       "          [ 2.1747e-01, -5.5122e-02, -2.2729e-01,  ..., -3.3487e-01,\n",
       "            5.7614e-01, -3.8694e-01],\n",
       "          [ 3.1471e-01, -7.6206e-02,  1.0735e-01,  ..., -3.5014e-01,\n",
       "            3.2091e-01, -3.5739e-01],\n",
       "          ...,\n",
       "          [ 1.3759e-01, -2.4057e-02, -2.3292e-01,  ..., -4.5908e-01,\n",
       "            2.2311e-01, -6.1929e-01],\n",
       "          [ 3.0267e-01, -1.9793e-01, -3.2124e-01,  ..., -7.1776e-01,\n",
       "            4.4381e-01, -6.6698e-01],\n",
       "          [ 4.3243e-01, -4.7029e-02, -1.9091e-02,  ..., -3.8013e-01,\n",
       "            6.5184e-01, -7.6008e-01]],\n",
       "\n",
       "         [[-2.7286e-01,  5.4990e-02, -4.5746e-01,  ..., -1.7381e-01,\n",
       "           -1.8635e-02, -3.7828e-01],\n",
       "          [-4.0514e-01,  2.4089e-01, -3.4257e-01,  ...,  1.3163e-01,\n",
       "            1.8355e-01, -2.1384e-01],\n",
       "          [-4.0188e-01,  3.5793e-02, -2.8199e-01,  ...,  1.6804e-01,\n",
       "            5.8308e-02, -3.4559e-01],\n",
       "          ...,\n",
       "          [-5.4478e-01,  1.3765e-01, -2.7699e-01,  ...,  3.5032e-01,\n",
       "           -9.1963e-02, -4.0222e-01],\n",
       "          [-4.7227e-01,  4.3055e-02, -1.8085e-01,  ...,  7.1014e-02,\n",
       "           -1.7961e-01, -3.1282e-01],\n",
       "          [-3.8232e-01,  3.6603e-01, -4.4386e-01,  ..., -1.1394e-01,\n",
       "           -2.9966e-01, -4.3051e-01]],\n",
       "\n",
       "         [[-2.6360e-01,  1.3743e-01,  2.9187e-01,  ...,  2.1083e-01,\n",
       "            3.1382e-01,  5.3461e-01],\n",
       "          [-4.4692e-01,  2.9673e-01,  3.6963e-01,  ...,  2.2120e-01,\n",
       "            2.8019e-01,  6.0401e-01],\n",
       "          [-2.9989e-01,  1.2915e-01,  4.9877e-01,  ...,  2.4780e-01,\n",
       "            1.1716e-01,  6.8828e-01],\n",
       "          ...,\n",
       "          [ 1.1086e-01, -1.9589e-01,  3.2770e-01,  ...,  1.9262e-01,\n",
       "            2.9159e-02,  6.1235e-01],\n",
       "          [-1.1959e-02, -9.8051e-02,  4.2960e-01,  ...,  2.4844e-01,\n",
       "            2.7309e-02,  5.9987e-01],\n",
       "          [ 9.7555e-02,  3.1115e-02,  5.2527e-01,  ...,  3.1809e-01,\n",
       "            1.8945e-01,  5.5182e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 3.6269e-01,  4.5794e-02,  8.5694e-01,  ..., -2.8150e-01,\n",
       "            6.3697e-01,  1.4626e-01],\n",
       "          [ 1.3827e-01,  2.4995e-01,  7.0073e-01,  ...,  7.1494e-02,\n",
       "            6.6857e-01, -1.0994e-01],\n",
       "          [ 5.6878e-02,  2.7654e-01,  7.4385e-01,  ..., -5.7883e-02,\n",
       "            6.2673e-01, -3.0501e-01],\n",
       "          ...,\n",
       "          [ 6.8351e-02,  1.9079e-01,  7.7504e-01,  ...,  8.4556e-02,\n",
       "            7.5165e-01,  1.1279e-02],\n",
       "          [ 1.8842e-01,  2.5775e-01,  6.2289e-01,  ..., -2.9823e-02,\n",
       "            6.8559e-01, -2.0885e-01],\n",
       "          [ 3.5035e-01,  2.5038e-01,  5.2403e-01,  ..., -2.9367e-02,\n",
       "            4.6815e-01, -9.1926e-02]],\n",
       "\n",
       "         [[ 7.5033e-01, -1.0463e+00,  2.9433e-02,  ..., -2.8486e-01,\n",
       "           -1.1636e+00,  2.9168e-01],\n",
       "          [ 3.3814e-01, -9.0982e-01, -7.0435e-02,  ..., -4.0565e-01,\n",
       "           -7.1877e-01,  1.0877e-01],\n",
       "          [ 3.3280e-01, -5.2484e-01, -4.6868e-02,  ..., -5.1282e-01,\n",
       "           -5.9183e-01,  4.4870e-02],\n",
       "          ...,\n",
       "          [ 2.8930e-01, -4.3036e-01, -4.5278e-01,  ..., -3.5935e-01,\n",
       "           -4.4293e-01,  1.7920e-01],\n",
       "          [ 4.6656e-01, -5.5979e-01, -9.1723e-02,  ..., -2.6114e-01,\n",
       "           -3.7516e-01,  3.3385e-02],\n",
       "          [ 2.6980e-01, -3.4968e-01, -2.3663e-01,  ..., -2.9986e-01,\n",
       "           -3.1621e-01,  7.6272e-02]],\n",
       "\n",
       "         [[ 1.9278e-01, -3.3078e-01, -1.2757e-01,  ...,  5.2762e-01,\n",
       "            4.7168e-01, -1.0996e-01],\n",
       "          [ 1.2039e-01, -2.2162e-01, -6.0906e-02,  ...,  1.3708e-01,\n",
       "            4.0902e-01, -6.8259e-02],\n",
       "          [ 7.9144e-02, -1.1947e-01, -1.4158e-01,  ...,  1.2180e-01,\n",
       "            2.7106e-01, -1.3360e-01],\n",
       "          ...,\n",
       "          [ 5.0196e-01, -3.3573e-01, -1.9318e-01,  ...,  1.8212e-01,\n",
       "            3.4116e-01, -9.3903e-02],\n",
       "          [ 2.6904e-01, -1.2735e-01, -1.4142e-01,  ...,  1.8475e-01,\n",
       "            1.6219e-01, -1.0083e-01],\n",
       "          [ 1.7768e-01, -1.3455e-01, -1.2735e-01,  ...,  2.6224e-01,\n",
       "            2.3173e-01, -2.6125e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.6023e-01, -4.2104e-01, -2.8515e-01,  ..., -8.0354e-01,\n",
       "            6.7926e-01, -7.9978e-01],\n",
       "          [-4.7582e-02, -1.9736e-01, -3.8237e-01,  ..., -7.0560e-01,\n",
       "            3.5649e-01, -8.0713e-01],\n",
       "          [ 1.3505e-01, -3.0030e-01, -1.0785e-01,  ..., -3.9035e-01,\n",
       "            3.1591e-01, -6.5513e-01],\n",
       "          ...,\n",
       "          [ 7.8378e-02, -1.3246e-01, -4.2092e-01,  ..., -5.9086e-01,\n",
       "            3.0975e-01, -5.3934e-01],\n",
       "          [ 1.0607e-01, -2.2999e-01, -2.9760e-01,  ..., -4.5209e-01,\n",
       "            3.3939e-01, -7.6361e-01],\n",
       "          [ 2.9097e-01, -2.4772e-01, -1.9826e-01,  ..., -5.0540e-01,\n",
       "            5.2105e-01, -5.8532e-01]],\n",
       "\n",
       "         [[-2.6924e-01,  2.8957e-02, -4.4644e-01,  ..., -2.1283e-01,\n",
       "           -1.2580e-02, -3.6509e-01],\n",
       "          [-5.3329e-01,  1.0148e-01, -4.5734e-01,  ..., -1.2311e-01,\n",
       "            9.2883e-02, -2.1155e-01],\n",
       "          [-2.8048e-01,  1.2088e-02, -2.4823e-01,  ...,  2.0531e-01,\n",
       "           -4.1112e-02, -3.3732e-01],\n",
       "          ...,\n",
       "          [-2.2530e-01,  9.6729e-02, -1.6781e-01,  ...,  1.3122e-01,\n",
       "           -1.0058e-01, -3.4202e-01],\n",
       "          [-3.9650e-01,  7.3285e-02, -3.0423e-01,  ...,  2.4637e-01,\n",
       "            1.0542e-01, -1.6368e-01],\n",
       "          [-3.3164e-01,  2.3919e-01, -4.5444e-01,  ..., -1.0609e-01,\n",
       "           -2.4153e-02, -3.9345e-01]],\n",
       "\n",
       "         [[-2.6491e-01,  1.3904e-01,  3.3067e-01,  ...,  1.9418e-01,\n",
       "            2.6161e-01,  5.5250e-01],\n",
       "          [-1.3142e-01,  5.3938e-02,  4.9435e-01,  ...,  3.8074e-01,\n",
       "            2.3678e-01,  7.0783e-01],\n",
       "          [ 8.3801e-02, -5.4110e-02,  2.9259e-01,  ...,  1.8127e-01,\n",
       "            1.2475e-01,  7.2434e-01],\n",
       "          ...,\n",
       "          [-1.4785e-01, -1.2070e-01,  6.8685e-01,  ...,  3.7602e-01,\n",
       "           -5.1070e-02,  8.3054e-01],\n",
       "          [ 1.2910e-01, -1.3193e-01,  3.6750e-01,  ...,  1.1228e-01,\n",
       "            1.1562e-01,  7.2051e-01],\n",
       "          [-8.0334e-02,  3.4776e-02,  6.6094e-01,  ...,  3.2663e-01,\n",
       "            1.5688e-01,  7.2247e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 3.4406e-01,  6.8143e-02,  8.2922e-01,  ..., -2.6480e-01,\n",
       "            6.4132e-01,  1.3542e-01],\n",
       "          [ 8.1183e-02,  2.1614e-01,  5.2831e-01,  ...,  2.6559e-02,\n",
       "            7.4832e-01,  1.4336e-03],\n",
       "          [ 5.3491e-02,  2.6417e-01,  4.9954e-01,  ...,  6.7356e-02,\n",
       "            6.8902e-01, -5.4571e-02],\n",
       "          ...,\n",
       "          [ 1.3963e-01,  2.6936e-01,  5.5752e-01,  ..., -7.5513e-02,\n",
       "            5.8597e-01, -2.7776e-01],\n",
       "          [ 2.2170e-01,  2.9675e-01,  4.7102e-01,  ...,  1.0624e-01,\n",
       "            6.8767e-01, -1.4477e-01],\n",
       "          [ 2.9108e-01,  3.6655e-01,  4.6963e-01,  ...,  3.1332e-02,\n",
       "            5.3931e-01, -5.0649e-02]],\n",
       "\n",
       "         [[ 7.3324e-01, -1.0492e+00,  4.8086e-02,  ..., -2.9212e-01,\n",
       "           -1.1700e+00,  3.0463e-01],\n",
       "          [ 5.1042e-01, -7.5074e-01, -3.5521e-03,  ..., -3.9265e-01,\n",
       "           -5.7140e-01,  6.6103e-02],\n",
       "          [ 4.0206e-01, -6.0236e-01, -1.2997e-02,  ..., -2.4592e-01,\n",
       "           -4.7813e-01,  6.6715e-02],\n",
       "          ...,\n",
       "          [ 2.9097e-01, -5.3034e-01,  7.9936e-02,  ..., -3.0588e-01,\n",
       "           -1.8933e-01,  1.4773e-01],\n",
       "          [ 2.6950e-01, -5.8839e-01, -1.0542e-01,  ..., -2.0273e-01,\n",
       "           -5.7679e-01,  1.2814e-01],\n",
       "          [ 4.7470e-01, -4.4882e-01, -8.8334e-02,  ..., -4.5571e-01,\n",
       "           -4.5543e-01,  3.6157e-02]],\n",
       "\n",
       "         [[ 1.8460e-01, -3.3858e-01, -1.5683e-01,  ...,  4.9892e-01,\n",
       "            4.5210e-01, -1.3175e-01],\n",
       "          [ 9.7540e-02, -2.3192e-01, -2.0650e-01,  ...,  2.4072e-01,\n",
       "            4.4687e-01, -8.6905e-02],\n",
       "          [ 1.5258e-01, -1.5007e-01, -1.4692e-01,  ...,  2.4812e-01,\n",
       "            2.4827e-01, -6.9500e-02],\n",
       "          ...,\n",
       "          [ 1.6446e-01, -1.4341e-01, -3.2371e-01,  ...,  2.2878e-01,\n",
       "            3.7990e-01, -9.9713e-02],\n",
       "          [ 9.0440e-02, -1.8855e-01, -4.3715e-02,  ...,  2.0498e-01,\n",
       "            3.4641e-01, -5.9393e-02],\n",
       "          [ 2.6242e-01, -2.6335e-01, -1.7450e-01,  ...,  4.1860e-01,\n",
       "            1.4080e-01, -2.8548e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.7132e-01, -3.9961e-01, -2.6460e-01,  ..., -8.2910e-01,\n",
       "            6.7508e-01, -7.8585e-01],\n",
       "          [ 1.9689e-01, -3.0063e-01, -2.3833e-01,  ..., -7.3320e-01,\n",
       "            4.4571e-01, -7.5566e-01],\n",
       "          [ 2.8589e-01, -2.4708e-01, -9.8223e-02,  ..., -6.2448e-01,\n",
       "            4.4967e-01, -7.0747e-01],\n",
       "          ...,\n",
       "          [ 2.5266e-01, -9.8012e-02, -1.2745e-01,  ..., -4.9795e-01,\n",
       "            4.6342e-01, -4.5581e-01],\n",
       "          [ 2.1766e-01, -2.7055e-01, -2.5147e-01,  ..., -6.9539e-01,\n",
       "            5.2100e-01, -6.1940e-01],\n",
       "          [ 5.7185e-01, -2.5231e-01, -2.0217e-01,  ..., -5.4447e-01,\n",
       "            4.6101e-01, -4.1106e-01]],\n",
       "\n",
       "         [[-2.7493e-01,  4.9830e-02, -4.4379e-01,  ..., -1.9181e-01,\n",
       "            8.6978e-04, -3.7589e-01],\n",
       "          [-3.8381e-01,  9.0584e-02, -3.2925e-01,  ...,  8.7710e-02,\n",
       "            2.7598e-02, -2.9641e-01],\n",
       "          [-3.0948e-01,  1.6903e-01, -2.3437e-01,  ..., -5.9992e-02,\n",
       "            1.2937e-02, -2.9024e-01],\n",
       "          ...,\n",
       "          [-2.2948e-01,  1.0356e-02, -1.8655e-01,  ...,  1.4035e-01,\n",
       "            2.4689e-02, -4.0447e-01],\n",
       "          [-3.8093e-01,  1.2597e-01, -5.7021e-01,  ..., -5.2836e-03,\n",
       "           -5.2741e-02, -2.5417e-01],\n",
       "          [-3.3078e-01,  4.3830e-01, -4.4261e-01,  ..., -1.3636e-01,\n",
       "           -1.4598e-01, -4.3383e-01]],\n",
       "\n",
       "         [[-2.7257e-01,  1.2325e-01,  3.0896e-01,  ...,  2.0423e-01,\n",
       "            2.5805e-01,  5.5392e-01],\n",
       "          [-5.7119e-02,  6.8277e-02,  3.9405e-01,  ...,  1.5172e-01,\n",
       "            6.9706e-02,  6.6117e-01],\n",
       "          [-1.2701e-01,  2.8241e-02,  5.1096e-01,  ...,  1.7629e-01,\n",
       "            7.7856e-02,  5.3786e-01],\n",
       "          ...,\n",
       "          [-5.3337e-02, -2.0448e-02,  3.4995e-01,  ...,  2.1914e-01,\n",
       "            2.7513e-01,  5.3131e-01],\n",
       "          [-1.5395e-01, -5.6935e-02,  6.6011e-01,  ...,  4.0422e-01,\n",
       "            1.0355e-01,  7.1750e-01],\n",
       "          [-1.0361e-01,  2.1366e-02,  5.6911e-01,  ...,  3.6314e-01,\n",
       "            5.3255e-01,  6.6373e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[ 1.2372e-01, -8.0341e-02,  1.2645e-01,  ...,  1.0515e-01,\n",
       "            1.0308e-01, -3.6344e-01],\n",
       "          [-2.0077e-03, -8.9920e-02, -2.3330e-01,  ...,  2.2151e-01,\n",
       "           -1.5354e-01,  3.3587e-01],\n",
       "          [ 1.0900e-01, -1.2313e-01, -2.1192e-01,  ...,  1.5514e-01,\n",
       "           -3.1971e-02,  1.5078e-01],\n",
       "          ...,\n",
       "          [-2.0177e-02,  4.8476e-02, -3.1886e-01,  ..., -2.2467e-01,\n",
       "           -2.5042e-02, -5.9066e-02],\n",
       "          [-8.2171e-02, -3.5509e-02, -2.8695e-01,  ...,  1.4745e-02,\n",
       "           -1.7887e-01,  2.1584e-02],\n",
       "          [-2.7790e-01, -1.0987e-01,  1.3986e-01,  ..., -2.8819e-01,\n",
       "            2.5340e-01, -7.8999e-02]],\n",
       "\n",
       "         [[ 4.7697e-01,  3.2814e-02, -2.5759e-01,  ...,  7.3226e-02,\n",
       "            6.9585e-02,  1.8402e-01],\n",
       "          [ 4.2409e-01, -6.8719e-02, -4.4506e-01,  ...,  2.7008e-01,\n",
       "            1.1189e-01,  6.4401e-02],\n",
       "          [ 4.7624e-01,  7.0228e-02, -3.5997e-01,  ...,  2.2712e-01,\n",
       "            1.0963e-01,  1.6736e-01],\n",
       "          ...,\n",
       "          [ 4.8269e-01, -2.0731e-01, -2.4419e-01,  ...,  2.0084e-02,\n",
       "           -1.2075e-01,  1.3605e-01],\n",
       "          [ 6.3834e-01, -1.8911e-01, -2.1757e-01,  ...,  4.0286e-02,\n",
       "            2.9961e-02,  1.2314e-01],\n",
       "          [ 2.1232e-01, -2.5883e-02, -4.9339e-02,  ...,  2.4829e-01,\n",
       "           -2.4334e-01,  7.0388e-02]],\n",
       "\n",
       "         [[ 2.2196e-01,  1.2276e-01,  6.8550e-02,  ...,  2.2578e-01,\n",
       "            1.0158e-01, -3.7854e-01],\n",
       "          [ 1.0151e-01, -1.4157e-01, -1.2793e-01,  ..., -3.4177e-02,\n",
       "           -1.4251e-01, -3.1446e-01],\n",
       "          [-4.0651e-02, -1.0704e-01, -1.7401e-01,  ...,  2.0793e-01,\n",
       "           -2.9925e-02, -3.1980e-01],\n",
       "          ...,\n",
       "          [ 2.4717e-01,  1.7881e-01, -1.5457e-01,  ...,  1.5435e-01,\n",
       "            1.6676e-01, -5.0785e-01],\n",
       "          [ 4.0867e-02,  8.1476e-02, -1.8892e-01,  ...,  6.1550e-02,\n",
       "           -3.0738e-02, -4.7763e-01],\n",
       "          [ 2.4089e-01, -1.0114e-01, -2.0375e-01,  ...,  4.9215e-01,\n",
       "            1.6443e-01, -2.2988e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.9893e-01, -1.0463e-01,  5.0403e-01,  ..., -5.4143e-02,\n",
       "            8.0244e-01, -3.3364e-02],\n",
       "          [ 2.0799e-01, -2.4041e-02,  6.6617e-01,  ..., -1.6258e-01,\n",
       "            1.0832e+00,  1.2002e-01],\n",
       "          [ 8.8371e-02, -1.2096e-01,  8.0918e-01,  ..., -2.6858e-02,\n",
       "            1.0273e+00,  7.5164e-02],\n",
       "          ...,\n",
       "          [ 1.6663e-01,  1.4123e-01,  5.0934e-01,  ..., -1.4610e-01,\n",
       "            1.1291e+00, -1.4425e-01],\n",
       "          [-7.7953e-04, -2.0459e-01,  4.9500e-01,  ...,  1.8907e-03,\n",
       "            1.0743e+00, -1.7390e-01],\n",
       "          [ 1.9123e-01,  1.8166e-01,  5.2118e-01,  ...,  4.1026e-02,\n",
       "            1.0733e+00, -3.3631e-01]],\n",
       "\n",
       "         [[-1.2874e-01,  2.0883e-01,  2.7282e-01,  ...,  4.3715e-02,\n",
       "           -8.1206e-01,  5.0336e-01],\n",
       "          [-1.1322e-01,  2.5995e-01,  3.9254e-01,  ...,  1.5898e-01,\n",
       "           -8.5837e-01,  6.7587e-01],\n",
       "          [ 1.2909e-01,  3.8916e-01,  3.9655e-01,  ...,  4.0745e-02,\n",
       "           -8.3141e-01,  6.3685e-01],\n",
       "          ...,\n",
       "          [ 1.8295e-01,  2.0657e-01,  1.4714e-01,  ..., -6.3845e-02,\n",
       "           -8.3312e-01,  2.0765e-01],\n",
       "          [ 1.1416e-01, -3.0936e-02,  1.4701e-01,  ..., -4.2563e-02,\n",
       "           -6.4103e-01,  2.3438e-01],\n",
       "          [ 7.6739e-02, -1.1963e-01,  9.7012e-02,  ..., -2.1913e-01,\n",
       "           -5.5106e-01,  4.2097e-01]],\n",
       "\n",
       "         [[-7.5624e-02,  5.1674e-03, -7.3891e-01,  ..., -2.1905e-01,\n",
       "           -1.2102e-01,  3.5186e-01],\n",
       "          [-5.1779e-02,  2.2075e-01, -8.3999e-01,  ...,  2.7224e-02,\n",
       "            2.8705e-01,  6.1645e-01],\n",
       "          [-5.4566e-02,  2.3115e-01, -1.0178e+00,  ...,  3.2560e-02,\n",
       "            3.4365e-01,  3.8287e-01],\n",
       "          ...,\n",
       "          [ 4.4424e-02,  6.0000e-02, -7.5210e-01,  ...,  1.0579e-01,\n",
       "            2.3547e-01,  3.2005e-01],\n",
       "          [-1.8957e-01,  1.0345e-01, -8.6132e-01,  ..., -1.8489e-02,\n",
       "            7.6295e-02,  5.1414e-01],\n",
       "          [-1.3298e-02,  2.7686e-01, -9.9655e-01,  ...,  8.4901e-02,\n",
       "            1.2100e-01,  1.2407e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.2009e-01, -1.1783e-01,  8.6621e-02,  ...,  1.1741e-01,\n",
       "            1.5330e-01, -3.6552e-01],\n",
       "          [-3.3734e-02,  3.0530e-02, -3.9961e-02,  ..., -1.0936e-01,\n",
       "           -1.4563e-01, -3.0415e-01],\n",
       "          [-1.2503e-01, -7.3122e-02, -1.6465e-01,  ...,  3.9031e-02,\n",
       "            1.8398e-01, -1.6305e-01],\n",
       "          ...,\n",
       "          [ 2.6903e-01, -1.2330e-01,  7.0587e-02,  ..., -1.3564e-01,\n",
       "            7.8118e-02, -4.0383e-02],\n",
       "          [-6.2160e-04, -4.7284e-02, -4.0819e-01,  ..., -7.5026e-02,\n",
       "            5.0575e-02, -1.4048e-02],\n",
       "          [-1.8255e-01, -6.5480e-03, -1.0494e-01,  ..., -2.2297e-01,\n",
       "            5.3026e-02, -1.1647e-01]],\n",
       "\n",
       "         [[ 5.2733e-01,  2.8028e-02, -2.6112e-01,  ...,  6.3775e-02,\n",
       "            6.9284e-02,  1.5301e-01],\n",
       "          [ 7.0668e-01,  1.9675e-01, -4.9049e-01,  ...,  1.7870e-01,\n",
       "           -8.3704e-02,  1.7757e-01],\n",
       "          [ 6.0039e-01, -1.3062e-01, -2.5531e-01,  ...,  1.2114e-01,\n",
       "           -1.4853e-01,  1.9138e-01],\n",
       "          ...,\n",
       "          [ 6.3204e-01, -1.4731e-01, -1.3156e-01,  ...,  2.5650e-01,\n",
       "           -1.3680e-02,  6.8905e-02],\n",
       "          [ 5.2744e-01, -6.6219e-02, -2.7952e-01,  ...,  1.6757e-01,\n",
       "           -6.2943e-02,  5.1618e-02],\n",
       "          [ 5.9884e-01, -4.8475e-02, -2.3326e-01,  ...,  8.8842e-02,\n",
       "           -1.6483e-01,  1.2343e-01]],\n",
       "\n",
       "         [[ 2.0060e-01,  1.4218e-01,  4.3514e-02,  ...,  2.3038e-01,\n",
       "            1.0941e-01, -4.2820e-01],\n",
       "          [ 1.3646e-01, -6.6512e-02, -1.4832e-02,  ...,  1.9052e-01,\n",
       "           -2.1917e-01, -3.6401e-01],\n",
       "          [ 1.0894e-01,  8.4077e-02, -1.2242e-01,  ...,  1.9353e-01,\n",
       "           -1.2773e-01, -4.7992e-01],\n",
       "          ...,\n",
       "          [ 2.5109e-01,  2.3351e-02,  8.3305e-02,  ...,  2.7634e-01,\n",
       "           -1.1215e-01, -3.3044e-01],\n",
       "          [-1.8657e-03,  7.3741e-02, -1.2652e-01,  ...,  2.0818e-01,\n",
       "            6.0180e-02, -4.4788e-01],\n",
       "          [-8.8902e-03, -1.0681e-01, -2.5784e-01,  ...,  2.1424e-01,\n",
       "            8.6685e-02, -3.7464e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.6007e-01, -5.4536e-02,  4.6918e-01,  ..., -4.2282e-02,\n",
       "            7.7120e-01, -1.1630e-02],\n",
       "          [ 2.9106e-02, -7.4611e-02,  4.6488e-01,  ...,  4.9434e-02,\n",
       "            9.2999e-01, -3.5040e-02],\n",
       "          [ 1.1442e-02,  3.5254e-02,  5.1541e-01,  ...,  4.0842e-02,\n",
       "            1.0156e+00,  3.8261e-02],\n",
       "          ...,\n",
       "          [ 2.0237e-01,  3.4399e-01,  3.7237e-01,  ..., -1.2802e-01,\n",
       "            1.1802e+00, -1.6801e-01],\n",
       "          [ 3.5070e-02,  1.1954e-01,  4.9268e-01,  ...,  1.5125e-01,\n",
       "            1.0312e+00, -9.5040e-02],\n",
       "          [ 4.7550e-03,  1.2084e-01,  6.3991e-01,  ..., -8.8026e-03,\n",
       "            7.7633e-01, -2.3518e-01]],\n",
       "\n",
       "         [[-1.3567e-01,  2.2703e-01,  2.3989e-01,  ...,  4.0018e-03,\n",
       "           -8.0697e-01,  5.5646e-01],\n",
       "          [-8.7027e-02,  3.5506e-01,  9.6927e-02,  ..., -5.6976e-03,\n",
       "           -7.9716e-01,  4.9948e-01],\n",
       "          [ 7.3131e-02,  1.8847e-01,  2.0477e-01,  ..., -4.7295e-02,\n",
       "           -6.9158e-01,  3.5836e-01],\n",
       "          ...,\n",
       "          [ 1.0572e-01,  1.3744e-01, -2.0248e-02,  ..., -2.5654e-01,\n",
       "           -8.3857e-01,  6.9468e-01],\n",
       "          [ 3.3654e-02,  1.6252e-01,  2.1153e-02,  ..., -6.6687e-02,\n",
       "           -8.3546e-01,  4.8658e-01],\n",
       "          [-1.1913e-01,  7.0700e-02,  1.9822e-01,  ..., -1.8908e-01,\n",
       "           -6.9519e-01,  2.5137e-01]],\n",
       "\n",
       "         [[-7.5332e-02, -8.8336e-03, -7.5694e-01,  ..., -1.9187e-01,\n",
       "           -1.3644e-01,  4.0254e-01],\n",
       "          [ 2.4832e-01,  2.2878e-01, -8.3910e-01,  ..., -1.6309e-01,\n",
       "            1.7250e-01,  8.6285e-02],\n",
       "          [-2.0984e-02,  1.8064e-01, -9.3641e-01,  ...,  2.7836e-02,\n",
       "            3.4398e-01,  3.7275e-01],\n",
       "          ...,\n",
       "          [ 1.6227e-01,  1.6857e-01, -8.1668e-01,  ..., -2.3077e-02,\n",
       "            4.7215e-02,  3.8839e-01],\n",
       "          [ 7.9763e-02,  7.0545e-02, -9.6520e-01,  ..., -8.2922e-02,\n",
       "            1.8773e-01,  5.7137e-01],\n",
       "          [-5.8669e-02,  1.1141e-01, -8.1699e-01,  ..., -9.1188e-02,\n",
       "           -8.1061e-02,  4.1049e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 1.5233e-01, -1.0541e-01,  8.1649e-02,  ...,  1.3231e-01,\n",
       "            1.1762e-01, -3.6703e-01],\n",
       "          [ 1.0051e-01, -5.9990e-02, -4.0956e-02,  ...,  8.0387e-02,\n",
       "            1.4397e-02, -1.2938e-01],\n",
       "          [-2.4586e-02, -1.1224e-01, -8.6781e-02,  ..., -6.3173e-02,\n",
       "           -3.6578e-02, -1.8723e-01],\n",
       "          ...,\n",
       "          [ 4.2536e-02,  3.9667e-03, -1.7889e-01,  ..., -7.5279e-02,\n",
       "           -1.4170e-01, -4.4718e-02],\n",
       "          [-1.2760e-03, -2.1891e-02, -6.3994e-02,  ..., -7.8112e-02,\n",
       "           -6.2395e-02, -1.3235e-01],\n",
       "          [-1.1013e-01,  5.3516e-02,  5.4700e-02,  ..., -2.3113e-01,\n",
       "            1.2714e-01, -2.0382e-01]],\n",
       "\n",
       "         [[ 5.1109e-01,  3.5963e-02, -2.6269e-01,  ...,  6.8246e-02,\n",
       "            7.9108e-02,  1.6728e-01],\n",
       "          [ 6.5624e-01, -8.5607e-02, -2.8941e-01,  ...,  5.6664e-02,\n",
       "           -7.1515e-02,  1.4837e-01],\n",
       "          [ 6.0842e-01, -5.8148e-02, -2.9700e-01,  ...,  1.9657e-01,\n",
       "           -3.2694e-02,  4.2651e-02],\n",
       "          ...,\n",
       "          [ 6.1336e-01, -1.5639e-01, -2.3378e-01,  ...,  1.8654e-01,\n",
       "           -2.1792e-01,  1.5783e-01],\n",
       "          [ 7.0117e-01, -1.7733e-03, -1.9228e-01,  ...,  8.6999e-02,\n",
       "            4.7699e-02,  1.5274e-01],\n",
       "          [ 4.4334e-01,  1.5372e-02, -2.1374e-01,  ...,  5.4315e-02,\n",
       "           -3.1573e-01, -8.9723e-02]],\n",
       "\n",
       "         [[ 1.7620e-01,  1.1332e-01,  7.0283e-02,  ...,  2.2954e-01,\n",
       "            9.2325e-02, -4.2854e-01],\n",
       "          [-1.5896e-02,  7.7854e-02, -3.2413e-02,  ...,  2.3308e-01,\n",
       "            4.0311e-02, -4.0749e-01],\n",
       "          [ 1.7118e-02,  2.0013e-01, -8.9401e-02,  ...,  3.5410e-01,\n",
       "           -4.3262e-02, -3.2996e-01],\n",
       "          ...,\n",
       "          [-1.2016e-01,  1.3398e-01,  2.5965e-02,  ...,  1.0274e-01,\n",
       "           -1.5991e-03, -3.7952e-01],\n",
       "          [-7.8341e-02,  2.3693e-01, -7.5548e-02,  ...,  1.1807e-01,\n",
       "           -2.4835e-02, -3.1581e-01],\n",
       "          [ 2.7937e-01, -2.6145e-01, -3.6913e-01,  ...,  1.1805e-01,\n",
       "            3.1328e-01, -3.3549e-01]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-1.8189e-01, -8.0296e-02,  4.9173e-01,  ..., -5.1877e-02,\n",
       "            7.6290e-01, -3.0417e-02],\n",
       "          [ 2.0035e-01, -1.3376e-01,  4.5455e-01,  ..., -2.4113e-02,\n",
       "            9.1788e-01, -9.8832e-02],\n",
       "          [-1.9508e-02, -1.5486e-01,  5.0887e-01,  ...,  9.6972e-02,\n",
       "            9.8462e-01,  3.7067e-02],\n",
       "          ...,\n",
       "          [ 1.7440e-01,  5.6688e-03,  3.8776e-01,  ..., -1.0471e-01,\n",
       "            8.9128e-01, -4.7591e-02],\n",
       "          [-6.8806e-02, -3.1400e-01,  5.9968e-01,  ...,  5.5096e-02,\n",
       "            9.6308e-01, -2.3272e-01],\n",
       "          [ 6.0113e-02, -2.5187e-02,  6.1023e-01,  ..., -2.7030e-01,\n",
       "            9.8841e-01, -3.8804e-01]],\n",
       "\n",
       "         [[-1.0368e-01,  2.1032e-01,  2.5745e-01,  ...,  2.3579e-02,\n",
       "           -8.0893e-01,  5.4093e-01],\n",
       "          [-9.1084e-02, -5.1443e-02,  1.6927e-01,  ..., -1.1418e-02,\n",
       "           -6.7748e-01,  2.5081e-01],\n",
       "          [ 7.6428e-02, -8.1450e-02,  2.1539e-01,  ..., -1.4601e-01,\n",
       "           -7.0541e-01,  3.3835e-01],\n",
       "          ...,\n",
       "          [ 4.0590e-02, -1.6907e-01,  2.2905e-01,  ..., -7.8483e-02,\n",
       "           -5.0831e-01,  4.3791e-01],\n",
       "          [-2.3781e-02,  9.4492e-03,  6.3670e-02,  ..., -1.6698e-01,\n",
       "           -7.1607e-01,  3.9632e-01],\n",
       "          [-7.9677e-02, -4.0350e-02,  3.2463e-01,  ..., -3.7132e-01,\n",
       "           -6.4634e-01,  4.3894e-02]],\n",
       "\n",
       "         [[-1.0457e-01, -9.5458e-03, -7.4350e-01,  ..., -2.0508e-01,\n",
       "           -1.5781e-01,  3.7238e-01],\n",
       "          [-2.0225e-01, -1.3693e-01, -8.8249e-01,  ..., -1.0173e-01,\n",
       "            1.8188e-01,  8.1938e-01],\n",
       "          [-2.8766e-01, -4.4890e-02, -7.1200e-01,  ..., -5.9838e-02,\n",
       "            6.9175e-02,  6.9796e-01],\n",
       "          ...,\n",
       "          [-2.2284e-02, -1.8384e-02, -6.7319e-01,  ..., -7.0774e-02,\n",
       "           -7.0355e-02,  2.7773e-01],\n",
       "          [-1.2174e-01, -5.3045e-02, -6.9043e-01,  ..., -1.0820e-01,\n",
       "            1.4073e-01,  4.7066e-01],\n",
       "          [-1.3515e-01,  1.0083e-01, -7.9556e-01,  ..., -1.4083e-01,\n",
       "            1.5364e-01,  2.7408e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-4.5055e-01, -5.3527e-01, -1.3043e-01,  ...,  8.6798e-01,\n",
       "            1.6443e+00,  1.6434e+00],\n",
       "          [-1.1486e+00, -6.6211e-01, -9.0728e-01,  ...,  3.0750e+00,\n",
       "            2.0437e+00,  2.2493e+00],\n",
       "          [-3.0129e-01,  3.2179e-01,  1.1390e-01,  ..., -1.6738e-01,\n",
       "            1.2561e+00,  1.1311e+00],\n",
       "          ...,\n",
       "          [-8.1443e-01, -1.0458e+00, -1.2100e-01,  ..., -8.5430e-01,\n",
       "            8.7251e-01,  1.0778e+00],\n",
       "          [-8.2781e-01, -1.4628e+00, -4.9459e-02,  ..., -5.9039e-01,\n",
       "            1.3710e+00,  1.0055e+00],\n",
       "          [-5.6266e-01,  6.9894e-01,  1.2733e+00,  ..., -7.3429e-01,\n",
       "           -3.2752e+00, -5.6577e-01]],\n",
       "\n",
       "         [[ 9.6256e-01, -5.4372e-01, -1.1179e+00,  ..., -1.6178e+00,\n",
       "           -1.8426e-01, -1.3012e-01],\n",
       "          [-1.5986e+00,  1.1787e+00, -1.5503e+00,  ...,  7.0935e-01,\n",
       "            1.7214e-01, -7.3269e-01],\n",
       "          [-1.1530e+00,  7.1594e-01,  6.0024e-01,  ...,  6.9251e-01,\n",
       "            1.6900e-01, -1.9785e+00],\n",
       "          ...,\n",
       "          [-1.9365e+00, -4.1399e-01,  2.4332e+00,  ..., -1.4862e+00,\n",
       "           -4.3062e-01,  6.5298e-01],\n",
       "          [ 1.9002e-01,  1.7241e+00,  1.4157e+00,  ..., -5.5546e-01,\n",
       "           -1.0393e+00, -1.3912e+00],\n",
       "          [ 6.4746e-01,  2.6689e-01,  8.3479e-02,  ..., -1.0010e+00,\n",
       "           -2.3288e+00, -1.1132e-01]],\n",
       "\n",
       "         [[ 1.3381e+00,  3.9018e-01, -1.3806e-01,  ...,  1.3614e+00,\n",
       "           -1.9496e+00,  2.5835e-01],\n",
       "          [ 6.3680e-01,  2.3462e-01,  2.5784e-01,  ...,  3.4187e+00,\n",
       "           -1.4246e+00,  1.4484e-01],\n",
       "          [ 7.4536e-01,  1.5015e-01, -7.1825e-01,  ...,  1.8580e+00,\n",
       "           -1.0018e+00, -2.8040e-01],\n",
       "          ...,\n",
       "          [ 8.7246e-01,  2.2212e+00,  1.9122e+00,  ..., -5.9359e-01,\n",
       "           -2.5031e+00, -1.3389e+00],\n",
       "          [ 3.2210e+00,  9.9360e-01, -1.5688e+00,  ..., -6.2517e-01,\n",
       "           -2.3097e+00, -9.5880e-01],\n",
       "          [-7.8975e-01,  2.1473e-01,  3.2601e-01,  ..., -4.1643e-01,\n",
       "           -1.7876e+00, -2.2501e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.0894e+00, -1.3048e+00, -1.0164e+00,  ...,  8.8481e-01,\n",
       "            2.4903e-01,  1.2317e+00],\n",
       "          [ 7.5221e-01,  5.8568e-01, -5.7634e-01,  ..., -1.9280e+00,\n",
       "           -1.6341e+00, -5.2251e-01],\n",
       "          [ 9.6589e-01, -1.3403e+00, -3.8222e-01,  ..., -3.7717e-01,\n",
       "           -9.8763e-01, -9.8593e-01],\n",
       "          ...,\n",
       "          [ 1.0102e+00, -3.1882e-01,  1.4448e-01,  ..., -8.9563e-02,\n",
       "            2.3836e-01, -1.0543e+00],\n",
       "          [ 1.2679e+00, -6.9511e-01,  2.9953e-01,  ...,  6.4756e-01,\n",
       "            6.6523e-01, -9.4700e-01],\n",
       "          [-5.3712e-01,  3.0350e-01, -1.7156e-01,  ...,  1.6045e-01,\n",
       "            3.2033e+00,  1.4868e+00]],\n",
       "\n",
       "         [[-7.6472e-01, -3.5924e-02,  1.1234e-01,  ...,  4.6731e-02,\n",
       "           -2.6501e+00, -2.0608e-01],\n",
       "          [-1.9984e+00,  1.2432e+00,  2.1600e+00,  ...,  2.5845e-01,\n",
       "           -3.2143e+00, -2.2897e+00],\n",
       "          [-1.4458e+00,  1.1514e+00,  4.3588e-01,  ...,  1.7621e+00,\n",
       "           -1.0454e+00,  6.4256e-01],\n",
       "          ...,\n",
       "          [-1.8601e+00,  1.2722e+00,  3.2135e+00,  ...,  1.3964e-01,\n",
       "            4.6792e-01, -2.0931e+00],\n",
       "          [ 4.0223e-01,  6.5593e-01,  1.7078e+00,  ..., -1.3918e-01,\n",
       "            6.8941e-01, -9.7197e-01],\n",
       "          [ 6.6747e-01, -3.0987e-01, -7.7529e-01,  ...,  7.2627e-01,\n",
       "            3.6178e-01,  6.5107e-01]],\n",
       "\n",
       "         [[-1.9689e+00,  2.6706e+00,  1.5299e+00,  ...,  7.4790e-01,\n",
       "           -8.1884e-01, -1.1263e+00],\n",
       "          [ 2.3260e+00, -5.1875e-01,  5.8660e+00,  ...,  1.3583e+00,\n",
       "            1.5749e+00, -2.4628e+00],\n",
       "          [ 8.1982e-01,  1.3772e+00,  4.2619e+00,  ..., -1.0761e-01,\n",
       "           -2.3218e-01, -7.6350e-01],\n",
       "          ...,\n",
       "          [ 3.8807e+00, -2.0164e+00,  2.7511e+00,  ...,  4.9928e-01,\n",
       "            1.9367e+00,  6.0046e-04],\n",
       "          [ 9.3145e-01,  8.7695e-01, -9.5381e-02,  ..., -8.5690e-01,\n",
       "           -7.9720e-02, -4.7966e-01],\n",
       "          [-1.1180e-01, -9.7718e-01, -1.7502e+00,  ..., -1.7777e+00,\n",
       "            4.7787e-01,  3.0672e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.0680e-01,  8.3449e-01, -1.4088e+00,  ...,  3.3088e-01,\n",
       "           -2.6312e+00, -3.2418e-01],\n",
       "          [-2.8224e-01, -1.0286e+00, -1.8145e-01,  ..., -1.2228e+00,\n",
       "            2.2303e-01,  1.8836e+00],\n",
       "          [ 1.2213e+00, -8.5824e-01, -6.2863e-01,  ..., -1.1544e+00,\n",
       "            1.2368e+00,  8.0500e-01],\n",
       "          ...,\n",
       "          [ 3.0897e-01,  1.8833e+00, -6.6536e-01,  ..., -4.0595e-02,\n",
       "           -3.9999e-01, -5.2846e-01],\n",
       "          [-1.6197e+00,  5.2453e-01, -7.7059e-01,  ...,  5.0636e-01,\n",
       "           -8.7717e-01, -8.0050e-01],\n",
       "          [-6.0120e-01,  7.3287e-01,  1.3054e+00,  ..., -7.6533e-01,\n",
       "           -3.3185e+00, -6.9487e-01]],\n",
       "\n",
       "         [[-3.1558e+00, -4.3331e-01, -7.9286e-01,  ..., -1.5925e+00,\n",
       "           -3.3588e+00,  2.9497e-01],\n",
       "          [-1.8860e+00, -5.1151e-01,  1.8149e+00,  ..., -9.2803e-01,\n",
       "            1.4567e+00, -1.6913e+00],\n",
       "          [ 1.0260e+00, -1.9996e+00, -4.7688e-02,  ...,  1.1035e+00,\n",
       "           -2.8840e-02,  7.4066e-02],\n",
       "          ...,\n",
       "          [ 3.5621e-01,  7.4911e-01, -2.9847e+00,  ..., -7.3666e-01,\n",
       "           -3.5566e+00, -8.7326e-01],\n",
       "          [-1.5694e+00,  1.1570e+00, -9.6196e-02,  ...,  1.0841e-01,\n",
       "           -3.7071e+00, -4.7693e-03],\n",
       "          [ 6.2158e-01,  3.0014e-01,  8.2817e-02,  ..., -9.5896e-01,\n",
       "           -2.3675e+00, -7.9399e-02]],\n",
       "\n",
       "         [[-5.8164e-01,  2.1249e+00, -9.3090e-01,  ...,  6.9360e-01,\n",
       "           -2.2764e+00, -1.5047e+00],\n",
       "          [-2.1613e-01,  1.0955e+00, -1.3971e+00,  ...,  1.0406e+00,\n",
       "           -5.9021e-02, -1.0862e+00],\n",
       "          [-1.1308e-01,  2.0377e+00, -1.2747e+00,  ...,  2.2614e+00,\n",
       "            3.0762e-01, -7.5349e-01],\n",
       "          ...,\n",
       "          [-1.7087e+00,  3.2066e+00,  2.6612e+00,  ..., -2.9445e-01,\n",
       "           -2.8973e+00,  1.3680e+00],\n",
       "          [ 2.2354e+00, -4.2391e-01,  1.4385e+00,  ..., -7.6845e-01,\n",
       "           -2.2227e+00,  1.4078e-01],\n",
       "          [-7.7643e-01,  1.8490e-01,  4.9373e-01,  ..., -4.7061e-01,\n",
       "           -1.7709e+00, -2.2359e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.8271e+00,  1.1735e+00, -4.3026e-01,  ..., -1.4084e+00,\n",
       "           -1.5211e+00,  1.1005e+00],\n",
       "          [ 2.0668e+00, -1.7577e+00,  9.0056e-01,  ...,  2.6639e-01,\n",
       "            8.6534e-01, -2.1033e+00],\n",
       "          [ 2.2740e+00, -2.3157e+00,  1.1535e+00,  ..., -1.2097e+00,\n",
       "            2.8619e-01, -1.9901e+00],\n",
       "          ...,\n",
       "          [-1.1377e-01,  7.1241e-01,  6.5038e-01,  ..., -1.0727e+00,\n",
       "           -1.6617e+00,  7.8599e-01],\n",
       "          [-4.0691e-01,  1.2519e+00,  1.2370e+00,  ..., -1.0004e+00,\n",
       "            1.7199e-01,  8.5910e-01],\n",
       "          [-6.2717e-01,  3.7086e-01, -9.7276e-02,  ...,  1.2510e-01,\n",
       "            3.2092e+00,  1.4433e+00]],\n",
       "\n",
       "         [[-1.6301e+00, -2.3162e-01, -1.2905e+00,  ...,  1.4609e+00,\n",
       "           -1.2447e+00, -2.6924e+00],\n",
       "          [-1.5636e+00,  1.4728e-01, -4.7821e-01,  ...,  2.8051e+00,\n",
       "            1.1737e+00, -8.9389e-01],\n",
       "          [-9.5143e-02,  1.0069e+00, -9.9335e-01,  ...,  2.2423e+00,\n",
       "            2.6700e-03, -4.9402e-01],\n",
       "          ...,\n",
       "          [-5.7612e-01,  1.1591e+00, -6.1022e-01,  ..., -1.7413e-01,\n",
       "           -3.1092e+00,  2.0276e+00],\n",
       "          [-1.5965e-01,  9.7220e-01, -2.5861e-02,  ...,  4.8556e-01,\n",
       "           -1.6212e+00,  6.9655e-01],\n",
       "          [ 6.9924e-01, -3.5387e-01, -7.9484e-01,  ...,  7.6552e-01,\n",
       "            2.4799e-01,  7.3986e-01]],\n",
       "\n",
       "         [[ 2.7446e+00, -1.3713e+00,  4.3627e-01,  ...,  3.6789e+00,\n",
       "            5.5705e-02,  4.7366e-01],\n",
       "          [ 1.8671e+00,  4.3125e-01,  3.2207e+00,  ..., -2.7196e-01,\n",
       "            2.8110e-01, -6.7273e-02],\n",
       "          [-2.0557e-01,  8.1900e-01,  1.1151e+00,  ...,  5.7660e-01,\n",
       "            7.5971e-01, -7.1824e-01],\n",
       "          ...,\n",
       "          [ 1.0455e+00,  7.3889e-01, -3.8944e-01,  ..., -1.1376e+00,\n",
       "            1.2662e+00, -7.9067e-01],\n",
       "          [ 2.1017e+00,  1.7458e+00, -6.9947e-01,  ..., -1.5667e+00,\n",
       "            1.5040e+00, -9.2767e-01],\n",
       "          [-7.2674e-02, -9.3879e-01, -1.7057e+00,  ..., -1.8334e+00,\n",
       "            5.4510e-01,  2.8026e-01]]],\n",
       "\n",
       "\n",
       "        [[[-2.0677e-01,  4.9325e-01,  1.2638e+00,  ..., -6.1028e-01,\n",
       "           -1.3823e+00, -8.3910e-01],\n",
       "          [-1.9522e+00, -1.5070e+00,  1.5253e+00,  ..., -5.5333e-01,\n",
       "            5.6643e-01, -5.1190e+00],\n",
       "          [-8.6065e-01,  2.3763e+00, -1.2674e-01,  ...,  9.4760e-01,\n",
       "           -1.2283e+00,  1.2209e-01],\n",
       "          ...,\n",
       "          [-5.9105e-01,  1.0232e-01,  9.8219e-01,  ..., -3.7589e-01,\n",
       "            8.8794e-01,  1.0002e+00],\n",
       "          [-7.7468e-01, -6.2345e-01,  6.9620e-01,  ...,  1.5532e+00,\n",
       "            1.3390e+00,  4.4590e-01],\n",
       "          [-5.9784e-01,  7.7418e-01,  1.2731e+00,  ..., -7.7041e-01,\n",
       "           -3.3459e+00, -6.5164e-01]],\n",
       "\n",
       "         [[-2.2147e+00,  1.8988e+00, -2.3210e+00,  ..., -2.5837e+00,\n",
       "           -2.2909e+00, -1.9705e+00],\n",
       "          [-1.6338e+00,  1.4535e+00, -1.9249e-01,  ..., -2.0602e+00,\n",
       "           -1.6623e+00, -1.2896e+00],\n",
       "          [-1.8902e-01, -8.4170e-01, -2.0435e+00,  ..., -2.9211e+00,\n",
       "            1.3738e+00, -1.4234e-01],\n",
       "          ...,\n",
       "          [-1.5654e+00,  1.3052e+00,  1.5431e-01,  ..., -2.0045e+00,\n",
       "            1.4521e+00,  2.1217e-02],\n",
       "          [-9.3449e-01,  1.3510e+00, -3.6573e-01,  ...,  5.0534e-01,\n",
       "            3.3407e-01, -1.9126e-01],\n",
       "          [ 5.7895e-01,  2.8561e-01,  7.1251e-02,  ..., -9.7904e-01,\n",
       "           -2.3505e+00, -1.1108e-01]],\n",
       "\n",
       "         [[ 2.3383e-01,  2.1229e+00, -9.8445e-01,  ...,  2.6423e+00,\n",
       "           -2.7492e+00, -1.0359e+00],\n",
       "          [ 1.2739e+00,  7.8450e-01, -3.5796e+00,  ...,  2.8235e-01,\n",
       "           -1.8756e+00, -1.1813e+00],\n",
       "          [-5.0057e-01, -7.2783e-01, -1.8154e+00,  ...,  2.3329e+00,\n",
       "           -2.5444e+00, -1.8949e+00],\n",
       "          ...,\n",
       "          [-7.3202e-01,  9.4479e-02,  7.8905e-01,  ..., -2.4556e+00,\n",
       "           -2.2228e+00, -1.9020e+00],\n",
       "          [ 1.3050e+00,  6.7200e-01,  2.5731e-01,  ..., -1.6199e+00,\n",
       "           -3.2651e+00, -9.8861e-01],\n",
       "          [-8.4460e-01,  1.8492e-01,  3.7863e-01,  ..., -4.4156e-01,\n",
       "           -1.8038e+00, -2.2487e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 1.9557e+00,  1.0021e+00,  8.5184e-01,  ...,  1.4332e-01,\n",
       "           -2.5502e+00, -7.1701e-01],\n",
       "          [ 1.7786e+00, -6.5784e-01, -5.6306e-01,  ..., -1.6210e+00,\n",
       "           -7.1376e-01, -1.4288e+00],\n",
       "          [-2.9863e-01,  8.0490e-01, -1.1035e+00,  ..., -1.2274e+00,\n",
       "            5.4274e-01, -2.8693e-01],\n",
       "          ...,\n",
       "          [-1.6490e-01,  1.9493e+00, -1.9375e+00,  ..., -7.9875e-02,\n",
       "            6.2474e-01, -2.3980e+00],\n",
       "          [ 4.0857e-01,  3.9175e-01, -1.4948e+00,  ...,  3.5512e-01,\n",
       "            6.4870e-01, -8.4664e-01],\n",
       "          [-5.0730e-01,  3.4662e-01, -1.8327e-01,  ...,  1.5143e-01,\n",
       "            3.1972e+00,  1.4852e+00]],\n",
       "\n",
       "         [[-1.5046e+00,  5.9314e-01, -4.3055e-01,  ...,  1.1090e+00,\n",
       "           -2.4152e+00, -3.9985e-01],\n",
       "          [-3.3496e-01,  2.7263e-01,  7.9516e-01,  ...,  2.2307e+00,\n",
       "            2.3814e-01, -1.1022e+00],\n",
       "          [-1.3328e+00, -2.7312e+00, -1.9505e+00,  ...,  4.8735e+00,\n",
       "           -1.1263e-01, -2.2544e-02],\n",
       "          ...,\n",
       "          [-5.2822e-01,  6.6929e-01,  3.9315e-02,  ...,  8.8482e-01,\n",
       "            1.8749e+00,  8.2361e-01],\n",
       "          [ 3.9177e-01, -2.3624e-01,  8.3704e-01,  ...,  8.0972e-01,\n",
       "            1.0927e+00,  6.9951e-01],\n",
       "          [ 6.4800e-01, -3.2202e-01, -8.2355e-01,  ...,  8.1563e-01,\n",
       "            2.9848e-01,  7.0697e-01]],\n",
       "\n",
       "         [[-6.5403e-01,  1.1073e+00, -7.4383e-01,  ..., -2.6304e-01,\n",
       "           -8.7598e-02, -1.3716e+00],\n",
       "          [ 2.0274e+00,  1.1251e+00, -1.9404e+00,  ...,  2.1413e+00,\n",
       "            3.7909e-01,  8.1213e-01],\n",
       "          [-1.0102e+00,  1.2294e+00,  1.6453e-01,  ...,  9.5499e-01,\n",
       "           -5.3518e-01,  9.6975e-01],\n",
       "          ...,\n",
       "          [ 1.6532e+00,  2.4686e+00, -1.6503e+00,  ...,  9.1186e-01,\n",
       "            1.1659e-01,  7.5998e-01],\n",
       "          [ 1.5560e+00,  1.4589e+00, -1.7076e+00,  ..., -9.8960e-01,\n",
       "           -7.9896e-01, -7.6735e-01],\n",
       "          [-9.4138e-02, -9.6424e-01, -1.7200e+00,  ..., -1.8001e+00,\n",
       "            4.7635e-01,  3.1975e-01]]]], grad_fn=<TransposeBackward0>), tensor([[[[-1.0196e+00,  4.8952e-01,  8.9452e-01,  ...,  2.6602e-01,\n",
       "            2.4801e+00,  1.9438e-01],\n",
       "          [ 1.0833e+00,  8.6254e-01, -8.8677e-01,  ..., -1.2578e+00,\n",
       "           -1.5353e-01,  1.5541e+00],\n",
       "          [-1.2486e-01,  8.5009e-01, -2.4801e-01,  ..., -2.2005e-01,\n",
       "           -9.0037e-01,  1.1304e+00],\n",
       "          ...,\n",
       "          [ 2.1475e+00,  4.2720e-01, -3.6799e-01,  ..., -5.0648e-01,\n",
       "           -6.1312e-01,  1.2480e+00],\n",
       "          [-3.3528e-01,  7.0419e-01, -4.0380e-01,  ...,  3.4838e-01,\n",
       "           -9.8086e-01,  8.2929e-01],\n",
       "          [-1.6430e-01,  3.8789e-01, -2.4169e-01,  ..., -1.9047e-01,\n",
       "            1.0415e+00,  2.9690e-01]],\n",
       "\n",
       "         [[-1.8759e+00, -2.7433e-01,  4.5292e-01,  ..., -2.0279e-01,\n",
       "           -1.1379e+00,  6.2917e-01],\n",
       "          [-1.2870e+00,  1.0417e+00, -2.9759e-01,  ..., -9.1868e-01,\n",
       "           -1.0451e+00,  1.4962e-01],\n",
       "          [-1.8205e+00,  7.5133e-03,  9.3302e-01,  ..., -3.1491e-01,\n",
       "           -1.1285e+00,  2.5909e-02],\n",
       "          ...,\n",
       "          [-4.7728e-01,  1.5488e+00,  8.7223e-01,  ..., -1.1384e+00,\n",
       "           -2.3783e+00,  7.0767e-02],\n",
       "          [ 6.6091e-01, -3.6647e-01,  5.7246e-01,  ..., -9.9091e-01,\n",
       "            6.2726e-01,  1.1298e+00],\n",
       "          [-3.2291e-01, -2.3082e-01,  2.2248e-01,  ...,  4.5919e-03,\n",
       "           -2.2128e-01, -1.0825e-01]],\n",
       "\n",
       "         [[ 3.3453e-01,  8.0366e-01,  7.7644e-02,  ..., -8.5109e-01,\n",
       "            1.3861e+00, -3.7584e-01],\n",
       "          [-6.2184e-01,  1.1310e+00,  6.0789e-02,  ..., -1.1180e+00,\n",
       "            7.7969e-01, -1.5188e+00],\n",
       "          [-8.7443e-01,  4.4349e-01,  1.7753e-01,  ...,  1.0985e-01,\n",
       "           -5.5676e-01, -5.7420e-01],\n",
       "          ...,\n",
       "          [-1.1877e+00,  2.0485e-01, -6.7014e-01,  ...,  5.9096e-01,\n",
       "           -1.6890e+00,  1.0549e+00],\n",
       "          [ 2.1135e-01, -3.8518e-01,  5.2402e-01,  ...,  8.2283e-01,\n",
       "           -9.1469e-01, -1.4414e-01],\n",
       "          [ 2.7318e-01, -3.5916e-01, -1.2959e-02,  ..., -6.3520e-03,\n",
       "           -7.7496e-01,  1.1181e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[-7.1660e-01,  1.3478e+00,  1.1182e-01,  ...,  2.9717e-01,\n",
       "            1.5743e-01, -6.9700e-01],\n",
       "          [ 8.8635e-01, -2.8714e-01,  3.4296e-01,  ..., -1.7811e+00,\n",
       "            7.3677e-01, -8.4729e-01],\n",
       "          [-3.6177e-01, -6.5879e-01,  3.3319e-01,  ..., -7.6468e-02,\n",
       "            1.7039e+00, -1.0643e+00],\n",
       "          ...,\n",
       "          [ 1.9394e+00, -1.8580e-01, -3.5411e-01,  ..., -1.2674e+00,\n",
       "            8.1662e-01, -6.4445e-01],\n",
       "          [ 5.7549e-01, -2.6893e-01,  1.6540e-01,  ..., -1.0910e+00,\n",
       "            7.7097e-03,  2.2794e-02],\n",
       "          [ 1.0591e+00,  2.5075e-01,  3.4982e-01,  ..., -1.6641e-01,\n",
       "           -8.9132e-02, -5.5612e-01]],\n",
       "\n",
       "         [[-1.2520e+00, -9.2982e-01, -8.2828e-02,  ..., -2.6920e+00,\n",
       "           -3.9256e-01, -6.5071e-01],\n",
       "          [-3.1372e-01, -6.4295e-01, -4.0641e-01,  ..., -2.5867e+00,\n",
       "           -2.4654e-01,  2.8282e-02],\n",
       "          [-8.3570e-01,  1.4867e+00, -2.2007e+00,  ..., -2.9879e+00,\n",
       "           -3.3284e-01,  4.4527e-01],\n",
       "          ...,\n",
       "          [-1.8875e-01, -1.0739e-01, -2.8416e-01,  ..., -1.9746e+00,\n",
       "            8.3822e-01,  4.2359e-01],\n",
       "          [-1.7953e-01, -2.0036e-02, -1.7246e+00,  ..., -1.6683e+00,\n",
       "           -2.5489e-01, -2.3858e-01],\n",
       "          [-3.8634e-01,  1.8676e-01,  3.8083e-02,  ..., -6.4461e-01,\n",
       "           -4.2534e-01,  2.7014e-01]],\n",
       "\n",
       "         [[ 1.2139e+00, -2.0324e+00,  1.1296e+00,  ..., -2.3619e+00,\n",
       "           -2.0091e-01, -1.7104e+00],\n",
       "          [-1.0150e-01, -5.1370e-01, -7.9144e-01,  ...,  1.5395e-01,\n",
       "           -1.0825e+00,  2.2968e-01],\n",
       "          [-1.6889e+00, -5.1510e-01,  6.7768e-01,  ...,  2.2278e-01,\n",
       "           -7.4711e-01, -1.7652e+00],\n",
       "          ...,\n",
       "          [ 3.5060e-01, -1.4493e+00,  2.7145e-01,  ..., -3.2827e-01,\n",
       "           -1.8777e+00,  2.2705e-01],\n",
       "          [-2.3080e-01,  1.8386e-01,  6.1029e-01,  ..., -2.8530e-01,\n",
       "           -2.2446e+00, -2.7996e-01],\n",
       "          [-7.4046e-01,  2.8253e-01, -8.4490e-02,  ...,  4.5410e-02,\n",
       "            2.0928e-01,  6.0502e-01]]],\n",
       "\n",
       "\n",
       "        [[[-1.2744e+00,  1.1839e+00, -4.7103e-01,  ..., -5.0835e-01,\n",
       "            2.7740e-01,  1.0895e+00],\n",
       "          [-1.4325e-02,  7.8750e-01,  9.9028e-02,  ...,  3.1650e-02,\n",
       "           -1.2131e+00,  9.4239e-01],\n",
       "          [-7.8461e-01, -2.4500e-01, -2.3792e-01,  ...,  1.0497e+00,\n",
       "            6.5524e-02, -1.7187e-01],\n",
       "          ...,\n",
       "          [ 1.8327e-01, -2.8535e-01,  8.2009e-01,  ...,  4.7469e-01,\n",
       "           -3.7597e-01, -4.0778e-02],\n",
       "          [-3.4911e-01,  1.0597e+00, -8.9417e-01,  ..., -2.3111e-01,\n",
       "            1.5394e-02,  5.5163e-01],\n",
       "          [-1.5348e-01,  4.2419e-01, -2.4118e-01,  ..., -2.0795e-01,\n",
       "            1.0521e+00,  3.1308e-01]],\n",
       "\n",
       "         [[-1.3649e+00,  1.4503e+00,  3.5077e-01,  ..., -3.9135e-01,\n",
       "           -1.1297e+00,  1.3621e+00],\n",
       "          [-1.5459e+00,  5.9090e-01,  1.3286e+00,  ..., -8.4839e-01,\n",
       "           -1.7422e+00, -3.5056e-01],\n",
       "          [-1.4479e+00, -4.7135e-01,  1.4264e+00,  ..., -9.2203e-01,\n",
       "           -1.0397e+00,  2.7456e+00],\n",
       "          ...,\n",
       "          [-7.8351e-01, -6.8371e-02,  1.0532e+00,  ...,  2.2858e-01,\n",
       "           -5.9165e-01,  1.2672e+00],\n",
       "          [ 3.4936e-01,  6.8521e-01,  7.1169e-01,  ..., -3.5904e-01,\n",
       "           -7.0865e-01, -1.7965e-01],\n",
       "          [-2.9701e-01, -2.0683e-01,  2.1298e-01,  ...,  1.6716e-02,\n",
       "           -2.5822e-01, -1.4409e-01]],\n",
       "\n",
       "         [[-7.5466e-01, -1.5548e-01, -2.4379e-01,  ...,  3.7798e-01,\n",
       "           -2.0944e-01, -5.9131e-01],\n",
       "          [-7.6848e-01,  1.0297e+00,  1.7759e-02,  ...,  7.2624e-01,\n",
       "           -1.1315e+00, -1.0123e+00],\n",
       "          [ 1.0311e+00, -2.1790e-01,  4.2051e-01,  ...,  6.9922e-01,\n",
       "            1.8946e+00, -1.6109e+00],\n",
       "          ...,\n",
       "          [-1.2283e-01,  8.3003e-02,  6.4756e-01,  ...,  1.4746e+00,\n",
       "            9.2437e-01, -9.3775e-01],\n",
       "          [-9.4340e-01, -8.1919e-01,  8.4674e-01,  ...,  7.6403e-01,\n",
       "           -5.4057e-01,  2.6497e-01],\n",
       "          [ 2.8163e-01, -3.3822e-01, -1.7083e-02,  ..., -9.9086e-03,\n",
       "           -7.5792e-01,  1.1262e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 2.6471e+00,  7.2850e-01,  1.2963e+00,  ..., -1.1689e+00,\n",
       "           -6.6712e-01,  8.5505e-01],\n",
       "          [-5.1053e-01, -3.9219e-01,  1.3533e+00,  ..., -9.7970e-02,\n",
       "            1.7138e+00, -2.9421e-02],\n",
       "          [-9.9320e-01, -4.2186e-01,  1.2014e+00,  ..., -4.8383e-01,\n",
       "           -5.3295e-01,  1.8340e-01],\n",
       "          ...,\n",
       "          [ 6.8346e-01, -3.0792e-01, -2.3485e-02,  ...,  1.3912e-01,\n",
       "           -7.4995e-02, -1.6683e+00],\n",
       "          [ 6.0618e-01, -6.3904e-02, -9.5710e-02,  ..., -4.5442e-01,\n",
       "           -1.1713e+00, -6.3620e-01],\n",
       "          [ 1.0501e+00,  2.1926e-01,  3.6425e-01,  ..., -1.5953e-01,\n",
       "           -6.6493e-02, -5.7443e-01]],\n",
       "\n",
       "         [[-5.8053e-01,  2.2509e+00,  3.5210e-01,  ..., -2.8559e+00,\n",
       "            2.0131e-01, -2.1756e+00],\n",
       "          [-4.4663e-01,  1.2566e+00, -2.1188e+00,  ..., -2.5980e+00,\n",
       "            2.6239e-02,  2.5837e-01],\n",
       "          [-6.2358e-02,  1.5344e+00, -1.0248e+00,  ..., -9.6996e-01,\n",
       "           -4.7872e-01,  9.5250e-01],\n",
       "          ...,\n",
       "          [-8.0909e-01,  1.8080e+00, -2.6378e-01,  ..., -2.8217e+00,\n",
       "           -3.4814e-01, -6.6035e-01],\n",
       "          [ 5.3203e-01,  1.9598e-01, -4.6121e-01,  ..., -1.5507e+00,\n",
       "            1.6424e-01, -1.2155e+00],\n",
       "          [-3.6902e-01,  2.1780e-01,  2.0237e-02,  ..., -6.9865e-01,\n",
       "           -4.0859e-01,  2.6045e-01]],\n",
       "\n",
       "         [[ 7.0327e-01, -1.4984e+00, -7.0159e-01,  ..., -4.1425e-01,\n",
       "           -1.6298e+00,  2.5972e+00],\n",
       "          [-9.6704e-01, -9.0079e-01,  3.5189e-01,  ..., -1.0234e+00,\n",
       "           -1.4576e+00, -1.0483e+00],\n",
       "          [-1.1229e+00, -1.2994e+00, -2.1896e+00,  ..., -1.4134e+00,\n",
       "            1.0414e+00,  3.1382e-01],\n",
       "          ...,\n",
       "          [ 1.4634e+00, -2.0239e+00, -2.9944e-01,  ..., -2.0004e-01,\n",
       "           -7.3513e-01,  2.3723e-01],\n",
       "          [ 4.2837e-01, -6.2406e-01, -1.0718e+00,  ..., -3.8931e-01,\n",
       "           -1.0112e+00, -3.3210e-01],\n",
       "          [-7.2541e-01,  2.6771e-01, -4.6483e-02,  ...,  2.0315e-02,\n",
       "            2.2548e-01,  6.0148e-01]]],\n",
       "\n",
       "\n",
       "        [[[ 3.1692e-01,  1.4718e+00,  9.1737e-01,  ..., -1.8650e+00,\n",
       "            2.2653e-01, -3.0369e-02],\n",
       "          [ 9.2133e-01,  6.9438e-01,  1.0991e+00,  ...,  1.1961e+00,\n",
       "           -8.4329e-01,  1.6654e+00],\n",
       "          [ 6.7918e-01,  1.2703e+00,  1.0801e+00,  ..., -5.8676e-01,\n",
       "            1.1834e-01, -2.9428e-02],\n",
       "          ...,\n",
       "          [-1.5721e+00,  2.1156e+00,  1.4916e+00,  ..., -1.6165e-01,\n",
       "            3.5771e-01,  1.4324e+00],\n",
       "          [-2.1152e-01,  1.5071e+00,  5.6263e-01,  ...,  2.2659e-01,\n",
       "           -5.5054e-02,  8.7098e-01],\n",
       "          [-1.8065e-01,  4.2891e-01, -2.4734e-01,  ..., -1.9720e-01,\n",
       "            1.0441e+00,  2.9160e-01]],\n",
       "\n",
       "         [[-1.0413e+00,  8.7255e-01, -7.4901e-01,  ...,  8.5208e-01,\n",
       "           -4.3167e-01,  2.1666e-01],\n",
       "          [ 2.1346e-01, -2.8645e-01,  9.0211e-01,  ...,  1.2711e+00,\n",
       "           -1.9537e+00,  7.2926e-01],\n",
       "          [-2.3448e-01, -1.0238e+00,  1.1347e+00,  ...,  2.2274e+00,\n",
       "           -9.8754e-03, -1.2744e+00],\n",
       "          ...,\n",
       "          [ 4.1605e-01, -8.2610e-02,  5.8452e-01,  ...,  3.0981e-01,\n",
       "           -6.4944e-01, -1.0264e+00],\n",
       "          [ 5.3355e-01,  2.1507e-01,  3.5888e-01,  ..., -2.2372e-01,\n",
       "            4.6695e-01,  6.2132e-01],\n",
       "          [-3.1329e-01, -2.2646e-01,  2.2900e-01,  ...,  5.6599e-02,\n",
       "           -2.5264e-01, -1.4648e-01]],\n",
       "\n",
       "         [[-2.3241e+00,  8.0845e-01, -9.0561e-02,  ...,  2.3352e+00,\n",
       "            1.0971e-01,  3.3283e-01],\n",
       "          [-8.1231e-01, -1.8923e-01,  5.5250e-01,  ..., -2.8088e-01,\n",
       "           -1.0635e+00,  4.2744e-01],\n",
       "          [ 6.2450e-01, -6.7350e-01,  1.8235e-01,  ..., -3.1009e-01,\n",
       "           -1.3909e+00, -6.8602e-01],\n",
       "          ...,\n",
       "          [ 8.5115e-01,  3.9448e-01, -6.1677e-01,  ...,  1.8053e+00,\n",
       "            2.0776e+00,  2.5106e-01],\n",
       "          [-1.7964e-01, -5.3297e-01,  3.2284e-01,  ...,  1.3627e+00,\n",
       "           -9.5943e-01,  9.1699e-02],\n",
       "          [ 2.7057e-01, -3.4054e-01, -2.8340e-02,  ..., -2.8451e-03,\n",
       "           -7.5512e-01,  1.0791e+00]],\n",
       "\n",
       "         ...,\n",
       "\n",
       "         [[ 5.5668e-01,  1.7980e-01, -1.6635e-01,  ..., -1.2522e+00,\n",
       "            2.5044e-02, -1.2451e+00],\n",
       "          [-4.5932e-01,  2.3140e-01,  1.8802e-01,  ..., -5.6659e-01,\n",
       "           -7.2071e-01, -6.7961e-01],\n",
       "          [-5.9705e-01, -2.3901e-01, -2.0430e-01,  ..., -9.7401e-01,\n",
       "           -4.0277e-01, -7.5038e-01],\n",
       "          ...,\n",
       "          [-1.2994e-01,  4.2110e-01, -9.9113e-02,  ..., -6.2312e-01,\n",
       "           -6.1411e-01, -1.2394e+00],\n",
       "          [ 6.5938e-02,  2.9552e-01,  1.1054e+00,  ..., -1.7816e-01,\n",
       "           -9.3824e-01, -2.0120e-01],\n",
       "          [ 1.0169e+00,  1.9740e-01,  3.5204e-01,  ..., -1.6315e-01,\n",
       "           -1.0138e-01, -5.5776e-01]],\n",
       "\n",
       "         [[-4.8021e-01,  1.4364e+00, -1.5354e+00,  ..., -3.6564e+00,\n",
       "           -1.2070e+00, -6.1690e-01],\n",
       "          [ 3.6372e-01, -9.6918e-01, -1.4405e+00,  ..., -2.7237e+00,\n",
       "           -7.5957e-01, -1.4217e+00],\n",
       "          [ 1.1659e+00,  8.8195e-01, -8.1152e-01,  ..., -1.0586e+00,\n",
       "            1.2443e+00, -4.0037e-01],\n",
       "          ...,\n",
       "          [ 7.0690e-01,  5.0554e-01, -1.3622e+00,  ..., -1.5866e+00,\n",
       "           -6.4869e-01, -1.1434e+00],\n",
       "          [-1.2931e+00,  9.9706e-02, -2.1616e-02,  ..., -1.5003e+00,\n",
       "           -3.0224e-01, -8.2825e-01],\n",
       "          [-3.7493e-01,  2.1792e-01,  3.6281e-02,  ..., -6.7124e-01,\n",
       "           -4.0612e-01,  2.6241e-01]],\n",
       "\n",
       "         [[-1.9500e-01, -1.1381e+00,  1.2315e-01,  ...,  3.6108e-02,\n",
       "           -1.4209e+00,  1.5447e+00],\n",
       "          [-1.3267e+00,  1.6821e-02, -5.0822e-01,  ..., -6.9060e-01,\n",
       "            3.9469e-01,  3.9208e-01],\n",
       "          [-9.3997e-02, -2.6776e-01,  5.7628e-01,  ...,  7.9660e-01,\n",
       "            9.4695e-01, -4.6747e-01],\n",
       "          ...,\n",
       "          [-7.4792e-01, -6.6410e-01, -1.3159e+00,  ..., -8.1311e-01,\n",
       "           -1.1348e+00, -7.4218e-01],\n",
       "          [ 5.9769e-01, -7.7536e-01, -4.5566e-01,  ..., -3.0059e-01,\n",
       "           -6.6285e-01,  7.9101e-01],\n",
       "          [-7.3559e-01,  2.8143e-01, -8.6784e-02,  ...,  3.2233e-02,\n",
       "            2.3140e-01,  6.1843e-01]]]], grad_fn=<TransposeBackward0>))), decoder_hidden_states=None, decoder_attentions=None, cross_attentions=None, encoder_last_hidden_state=tensor([[[-4.5522e-01, -4.8324e-01, -1.1120e+00,  ...,  1.1293e+00,\n",
       "          -3.3070e-01, -8.3485e-01],\n",
       "         [ 2.9534e-01, -2.3726e-01, -1.6786e-01,  ...,  2.0579e+00,\n",
       "           1.0986e+00, -5.1539e-01],\n",
       "         [ 4.5135e-01,  9.6353e-01, -3.8798e-01,  ...,  4.3344e-01,\n",
       "          -5.8341e-01, -5.9671e-01],\n",
       "         ...,\n",
       "         [ 5.5989e-02,  5.0441e-01,  2.5179e-01,  ..., -4.3534e-01,\n",
       "           4.0817e-02, -1.6295e-01],\n",
       "         [-4.3622e-01,  3.1871e+00,  6.9795e-01,  ...,  3.9164e-01,\n",
       "          -1.7549e+00, -1.6478e-01],\n",
       "         [ 1.0599e-02, -7.0050e-02, -4.4626e-02,  ..., -7.0160e-02,\n",
       "          -1.9263e-01, -7.3043e-01]],\n",
       "\n",
       "        [[ 2.7403e-01, -2.2729e-01,  2.1779e-01,  ...,  8.6530e-02,\n",
       "          -7.4091e-01,  2.8783e-01],\n",
       "         [ 6.8476e-02,  8.0827e-01, -6.7014e-02,  ..., -3.2850e-01,\n",
       "          -1.6333e+00,  3.8065e-01],\n",
       "         [ 9.0258e-01, -2.6485e-01, -7.3462e-04,  ..., -1.9850e-01,\n",
       "          -3.1904e-01, -1.3708e-01],\n",
       "         ...,\n",
       "         [-5.4099e-01, -2.8529e-01,  1.2922e-01,  ..., -1.8677e-01,\n",
       "          -1.1157e+00,  7.7572e-01],\n",
       "         [-1.1195e+00,  6.4328e-01,  2.8259e-01,  ...,  2.3852e-01,\n",
       "          -1.0032e+00,  7.1042e-01],\n",
       "         [-4.3798e-02, -8.0766e-02, -4.9749e-02,  ..., -6.3174e-02,\n",
       "          -2.1874e-01, -6.8360e-01]],\n",
       "\n",
       "        [[ 1.0861e-01, -2.2082e+00,  4.1419e-02,  ..., -9.1324e-01,\n",
       "          -4.9853e-01, -4.2681e-01],\n",
       "         [-1.3265e+00,  9.0085e-01, -5.1640e-02,  ..., -2.9749e-02,\n",
       "          -8.0917e-01,  1.8483e+00],\n",
       "         [ 2.6670e-01, -1.3010e+00, -1.3083e-02,  ...,  1.2227e-01,\n",
       "          -6.0922e-01,  1.3387e+00],\n",
       "         ...,\n",
       "         [-3.4986e-01,  1.1701e+00,  5.3957e-01,  ..., -1.0294e-01,\n",
       "          -1.3825e+00,  9.9322e-01],\n",
       "         [-5.5820e-01,  4.8946e-01,  5.5052e-01,  ..., -7.7154e-01,\n",
       "          -2.5348e-01,  8.6497e-01],\n",
       "         [-3.5234e-02, -6.6840e-02,  2.6129e-04,  ..., -1.2337e-01,\n",
       "          -1.5805e-01, -7.1576e-01]]], grad_fn=<MulBackward0>), encoder_hidden_states=None, encoder_attentions=None)"
      ]
     },
     "execution_count": 42,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "o = model.model(**b)\n",
    "o"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 48,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'<extra_id_99>�<extra_id_98>��<extra_id_98>��<extra_id_98> wilayah<extra_id_97><extra_id_97><extra_id_97> bahasa<extra_id_96> bahasa<extra_id_96>,<extra_id_96> orang bahasa<extra_id_95>angan)<extra_id_95>) yang\" dan men<extra_id_93>\" ialah<extra_id_92> menjadi dalam<extra_id_91> bahasa yang terdiri<extra_id_90> salah satu<extra_id_90> (<extra_id_89> bahasa<extra_id_88> bahasa Inggeris ( bahasa Melayu<extra_id_88> bahasa bahasa bahasa Melayu<extra_id_86> bahasa<extra_id_85><extra_id_85> \"bahasa<extra_id_84>c bahasa<extra_id_83>ak<extra_id_83> Melayu).). dan ditut<extra_id_82> bahasa<extra_id_81>ong R<extra_id_80>or<extra_id_79>. Di Malaysia<extra_id_78><extra_id_78>orang Melayu<extra_id_77>. Di<extra_id_76><extra_id_76> Johor<extra_id_75> dan bahasaezakan bahasa bahasa bahasa<extra_id_74>bahasa<extra_id_74>en<extra_id_73><extra_id_73> lain<extra_id_72>-</s>'"
      ]
     },
     "execution_count": 48,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "tokenizer.decode(o.logits.argmax(-1)[0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'<extra_id_99>�ꤼ ꤸ<extra_id_98> bawah keluarga bahasa<extra_id_97> bahasa rasmi di Brunei,<extra_id_96> penutur (seramai 260<extra_id_95>)<extra_id_94>\" untuk pent<extra_id_93>\" atau<extra_id_92> digunakan mewakili<extra_id_91> waktu yang sama merupakan salah satu<extra_id_90> (<extra_id_89> lain ialah bahasa Inggeris, bahasa Cina<extra_id_88> rasmi<extra_id_87> bahasa Melayu<extra_id_86> bahasa Indonesia<extra_id_85> \"bahasa<extra_id_84>bentuk vernak<extra_id_83> Melayu tempatan) yang ditut<extra_id_82> asli<extra_id_81>ulauan R<extra_id_80>or<extra_id_79>. Di selatan Thailand<extra_id_78>orang dari<extra_id_77>. Bahasa Melayu<extra_id_76> Melaka<extra_id_75> untuk membezakannya daripada bahasa-bahasa<extra_id_74>nolog<extra_id_73> terpisah<extra_id_72>-</s>'"
      ]
     },
     "execution_count": 49,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "tokenizer.decode(b['labels'][0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}
